Searched defs:eigen_device (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | cwise_ops_common.h | 101 const Device& eigen_device = ctx->eigen_device<Device>(); variable 109 eigen_device, out_flat, in0.template flat<Tin>(), 114 eigen_device, out_flat, in0.template scalar<Tin>(), 118 eigen_device, out_flat, in0.template flat<Tin>(), 123 eigen_device, out->shaped<Tout, 2>(bcast->result_shape()), 130 eigen_device, out->shaped<Tout, 3>(bcast->result_shape()), 137 eigen_device, out->shaped<Tout, 4>(bcast->result_shape()), 144 eigen_device, out->shaped<Tout, 5>(bcast->result_shape()), 178 const Device& d = context->eigen_device<Devic 207 const Device& eigen_device = ctx->eigen_device<Device>(); variable [all...] |
H A D | quantization_utils_test.cc | 34 void TestRequantizeMany(Eigen::ThreadPoolDevice* eigen_device, float input_min, argument 52 if (eigen_device == nullptr) { 59 *eigen_device, i_tensor, input_min, input_max, output_min, output_max, 112 // If eigen_device is NULL, then the reference implementation is tested. 114 Eigen::ThreadPoolDevice* eigen_device) { 137 TestRequantizeMany(eigen_device, r[0], r[1], r[2], r[3], 152 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -1.0f, 1.0f, vals); 153 TestRequantizeMany(eigen_device, -255.0f, 255.0f, -255.0f, 255.0f, vals); 154 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -12345678.0f, 12345678.0f, 156 TestRequantizeMany(eigen_device, 113 TestRequantizeManyInNewRange32To8Bit( Eigen::ThreadPoolDevice* eigen_device) argument 339 TestFloatToQuantizedInPlaceUsingEigen( Eigen::ThreadPoolDevice* eigen_device) argument 383 TestQuantizedToFloatInPlaceUsingEigen( Eigen::ThreadPoolDevice* eigen_device) argument [all...] |
/external/tensorflow/tensorflow/core/framework/ |
H A D | op_kernel.cc | 1215 const Eigen::ThreadPoolDevice& OpKernelContext::eigen_device() const { function in class:tensorflow::OpKernelContext 1220 const Eigen::GpuDevice& OpKernelContext::eigen_device() const { function in class:tensorflow::OpKernelContext 1226 const Eigen::SyclDevice& OpKernelContext::eigen_device() const { function in class:tensorflow::OpKernelContext
|
Completed in 127 milliseconds