| /external/tensorflow/tensorflow/core/kernels/ |
| 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<Device>()
207 const Device& eigen_device = ctx->eigen_device<Device>(); variable
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| 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
[all...] |
