Searched defs:eigen_device (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcwise_ops_common.h101 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 Dquantization_utils_test.cc34 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 Dop_kernel.cc1215 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