/external/tensorflow/tensorflow/core/kernels/ |
D | aggregate_ops.cc | 84 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); in Compute() 92 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); in Compute() 97 functor3(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2)); in Compute() 102 functor4(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 108 functor5(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 114 functor6(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 120 functor7(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 126 functor8(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 133 functor9(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute() 142 functor8p(ctx->template eigen_device<Device>(), To, I(r), I(r + 1), in Compute()
|
D | quantization_utils_test.cc | 34 void TestRequantizeMany(Eigen::ThreadPoolDevice* eigen_device, float input_min, in TestRequantizeMany() argument 52 if (eigen_device == nullptr) { in TestRequantizeMany() 59 *eigen_device, i_tensor, input_min, input_max, output_min, output_max, in TestRequantizeMany() 114 Eigen::ThreadPoolDevice* eigen_device) { in TestRequantizeManyInNewRange32To8Bit() argument 137 TestRequantizeMany(eigen_device, r[0], r[1], r[2], r[3], in TestRequantizeManyInNewRange32To8Bit() 152 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -1.0f, 1.0f, vals); in TestRequantizeManyInNewRange32To8Bit() 153 TestRequantizeMany(eigen_device, -255.0f, 255.0f, -255.0f, 255.0f, vals); in TestRequantizeManyInNewRange32To8Bit() 154 TestRequantizeMany(eigen_device, -1.0f, 1.0f, -12345678.0f, 12345678.0f, in TestRequantizeManyInNewRange32To8Bit() 156 TestRequantizeMany(eigen_device, -1.0f, 12345678.0f, -12345678.0f, in TestRequantizeManyInNewRange32To8Bit() 171 TestRequantizeMany(eigen_device, in_min, in_max, out_min, out_max, vals); in TestRequantizeManyInNewRange32To8Bit() [all …]
|
D | relu_op.h | 40 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 88 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate() 99 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 130 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate() 146 functor({context->eigen_device<Device>(), input.flat<T>(), alpha_, in Operate() 191 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), alpha, in OperateNoTemplate() 202 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 233 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate() 244 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 275 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate()
|
D | cwise_ops_common.h | 91 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local 103 eigen_device, out->template flat<Tout>(), in Compute() 117 eigen_device, out->template flat<Tout>(), in Compute() 130 eigen_device, out->template flat<Tout>(), in Compute() 150 functor::SetOneFunctor<Device, bool>()(eigen_device, in Compute() 153 functor::SetZeroFunctor<Device, bool>()(eigen_device, in Compute() 172 eigen_device, out_flat, in0.template flat<Tin>(), in Compute() 177 eigen_device, out_flat, in0.template scalar<Tin>(), in Compute() 181 eigen_device, out_flat, in0.template flat<Tin>(), in Compute() 186 eigen_device, out->shaped<Tout, 2>(bcast.result_shape()), in Compute() [all …]
|
D | conditional_accumulator.h | 94 accum_grad_->flat<T>().device(ctx->template eigen_device<Device>()) = in AllocateAndAssignToAccumGradFunction() 100 accum_grad_->flat<T>().device(ctx->template eigen_device<Device>()) += in AddToAccumGradFunction() 109 ctx->template eigen_device<Device>()) = in DivideAccumGradByCounter()
|
D | cudnn_pooling_gpu.cc | 54 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute() 126 context->eigen_device<GPUDevice>(), in Compute() 190 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute() 195 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute() 200 context->eigen_device<GPUDevice>(), out_backprop.tensor<T, 5>(), in Compute() 271 context->eigen_device<GPUDevice>(), in Compute()
|
D | spacetodepth_op.cc | 136 functor(context->eigen_device<Device>(), Tinput_v, block_size_, in Compute() 141 functor(context->eigen_device<Device>(), input.tensor<RT, 4>(), in Compute() 146 functor(context->eigen_device<Device>(), input.tensor<RT, 4>(), in Compute() 153 functor(context->eigen_device<Device>(), input.tensor<T, 4>(), in Compute()
|
D | transpose_op.cc | 193 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose() 202 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<CPUDevice>(), in, in DoTranspose() 225 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm, in TF_CALL_ALL_TYPES() 233 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<GPUDevice>(), in, in DoTranspose()
|
D | quantize_and_dequantize_op.cc | 116 f(ctx->eigen_device<Device>(), input.flat<T>(), signed_input_, num_bits_, in Compute() 121 f(ctx->eigen_device<Device>(), in Compute() 186 f(ctx->eigen_device<Device>(), gradient.template flat<T>(), in Compute() 193 f(ctx->eigen_device<Device>(), in Compute() 275 f(ctx->eigen_device<Device>(), input.flat<T>(), signed_input_, in Compute() 280 f(ctx->eigen_device<Device>(), in Compute() 330 functor(ctx->eigen_device<Device>(), input.flat<T>(), signed_input_, in Compute()
|
D | bias_op.cc | 141 const Device& d = context->eigen_device<Device>(); in Compute() 150 const Device& d = context->eigen_device<Device>(); in Compute() 159 const Device& d = context->eigen_device<Device>(); in Compute() 197 functor(ctx->eigen_device<Device>(), input.tensor<T, Dims>(), bias.vec<T>(), in Compute() 266 redux(context->eigen_device<Device>(), three_dims, output_backprop, in Compute() 275 redux(context->eigen_device<Device>(), two_dims, output_backprop, in Compute() 333 BiasGPU<T>::compute(context->template eigen_device<Device>(), in Compute() 462 BiasGradGPU<T>::compute(context->template eigen_device<Device>(), in ComputeWithCustomKernel()
|
D | fake_quant_ops.cc | 89 functor(context->eigen_device<Device>(), input.flat<float>(), min_, max_, in Operate() 138 functor(context->eigen_device<Device>(), gradient.flat<float>(), in OperateNoTemplate() 212 functor(context->eigen_device<Device>(), input.flat<float>(), in Compute() 263 functor(context->eigen_device<Device>(), gradient.flat<float>(), in Compute() 350 functor(context->eigen_device<Device>(), input.flat_inner_dims<float, 2>(), in Compute() 410 context->eigen_device<Device>(), gradient.flat_inner_dims<float, 2>(), in Compute()
|
D | pooling_ops_common.cc | 250 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute() 340 context->eigen_device<GPUDevice>(), in Compute() 415 context->eigen_device<Device>(), in Compute() 518 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute() 527 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute() 532 context->eigen_device<Device>(), out_backprop.tensor<T, 4>(), in Compute() 647 context->eigen_device<GPUDevice>(), in Compute() 736 context->eigen_device<GPUDevice>(), in Compute() 751 context->eigen_device<Device>(), in Compute()
|
D | depthtospace_op.cc | 121 functor(context->eigen_device<Device>(), Tinput_v, block_size_, in Compute() 126 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute() 136 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
|
D | fused_batch_norm_op.cc | 139 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()() 156 const CPUDevice& d = context->eigen_device<CPUDevice>(); in operator ()() 221 context->eigen_device<CPUDevice>(), transformed_y, perm, y_output); in operator ()() 277 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()() 293 const CPUDevice& d = context->eigen_device<CPUDevice>(); in operator ()() 332 context->eigen_device<CPUDevice>(), transformed_y, perm, y_output); in operator ()() 374 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()() 378 context->eigen_device<CPUDevice>(), x_input, in operator ()() 404 const CPUDevice& d = context->eigen_device<CPUDevice>(); in operator ()() 508 context, ::tensorflow::DoTranspose(context->eigen_device<CPUDevice>(), in operator ()() [all …]
|
D | strided_slice_op_impl.h | 96 context->eigen_device<Device>(), in HandleStridedSliceCase() 109 context->eigen_device<Device>(), in HandleStridedSliceCase() 136 context->eigen_device<Device>(), result->bit_casted_tensor<Proxy, NDIM>(), in HandleStridedSliceGradCase() 157 context->eigen_device<Device>(), result->bit_casted_tensor<Proxy, NDIM>(), in operator() 176 context->eigen_device<Device>(), in operator()
|
D | tensor_array.cc | 36 add_functor(ctx->template eigen_device<Device>(), sum->flat<T>(), \ 60 set_zero_functor(ctx->template eigen_device<Device>(), value->flat<T>()); \
|
D | diag_op_gpu.cu.cc | 62 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()() 98 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()()
|
D | stateless_random_ops.cc | 113 context, context->eigen_device<Device>(), /*key=*/nullptr, in Fill() 150 context, context->eigen_device<Device>(), /*key=*/nullptr, in Fill() 170 context, context->eigen_device<Device>(), /*key=*/nullptr, in Fill() 196 ctx, ctx->eigen_device<CPUDevice>(), rate_flat, num_rate, in Fill()
|
D | softplus_op.cc | 41 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 75 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate()
|
D | softsign_op.cc | 41 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate() 76 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate()
|
D | dense_update_ops.cc | 40 copy(context->eigen_device<Device>(), lhs->flat<T>(), rhs.flat<T>()); in Copy() 81 update_functor(context->template eigen_device<Device>(), Tparams.flat<T>(), in DoUpdate()
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | resize_nearest_neighbor_op.cc | 71 context->eigen_device<Device>(), input_data, st.height_scale, in Compute() 77 context->eigen_device<Device>(), input_data, st.height_scale, in Compute() 85 context->eigen_device<Device>(), input_data, st.height_scale, in Compute() 91 context->eigen_device<Device>(), input_data, st.height_scale, in Compute() 262 context->eigen_device<Device>(), input_data, height_scale, in Compute() 268 context->eigen_device<Device>(), input_data, height_scale, in Compute() 277 context->eigen_device<Device>(), input_data, height_scale, in Compute() 284 context->eigen_device<Device>(), input_data, height_scale, in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | device_base_test.cc | 31 Eigen::ThreadPoolDevice eigen_device(pool.AsEigenThreadPool(), in TEST() local 34 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
|
/external/tensorflow/tensorflow/core/kernels/linalg/ |
D | svd_op_gpu.cu.cc | 181 const GPUDevice& d = context->eigen_device<GPUDevice>(); in RunSVD() 241 const GPUDevice& d = context->eigen_device<GPUDevice>(); in RunSVD() 257 auto device = context->eigen_device<GPUDevice>(); in RunSVD() 305 auto device = context->eigen_device<GPUDevice>(); in PerformSVD_MgeqN() 333 const GPUDevice& d = context->eigen_device<GPUDevice>(); in PerformSVD_MlessN() 400 auto device = context->eigen_device<GPUDevice>(); in ComputeAsync()
|
/external/tensorflow/tensorflow/core/util/ |
D | tensor_ops_util.h | 45 out->flat<dtype>().device(ctx->eigen_device<Device>()) = \ in ZerosLikeTensor() 105 out->flat<dtype>().device(ctx->eigen_device<Device>()) = \ in BinaryAddTensors()
|