Home
last modified time | relevance | path

Searched refs:eigen_device (Results 1 – 25 of 172) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc84 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()
Dquantization_utils_test.cc34 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 …]
Drelu_op.h40 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()
Dcwise_ops_common.h91 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 …]
Dconditional_accumulator.h94 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()
Dcudnn_pooling_gpu.cc54 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()
Dspacetodepth_op.cc136 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()
Dtranspose_op.cc193 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()
Dquantize_and_dequantize_op.cc116 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()
Dbias_op.cc141 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()
Dfake_quant_ops.cc89 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()
Dpooling_ops_common.cc250 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()
Ddepthtospace_op.cc121 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()
Dfused_batch_norm_op.cc139 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 …]
Dstrided_slice_op_impl.h96 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()
Dtensor_array.cc36 add_functor(ctx->template eigen_device<Device>(), sum->flat<T>(), \
60 set_zero_functor(ctx->template eigen_device<Device>(), value->flat<T>()); \
Ddiag_op_gpu.cu.cc62 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()()
98 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()()
Dstateless_random_ops.cc113 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()
Dsoftplus_op.cc41 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate()
75 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate()
Dsoftsign_op.cc41 functor(context->eigen_device<Device>(), input.flat<T>(), in Operate()
76 functor(context->eigen_device<Device>(), g.flat<T>(), a.flat<T>(), in OperateNoTemplate()
Ddense_update_ops.cc40 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/
Dresize_nearest_neighbor_op.cc71 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/
Ddevice_base_test.cc31 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/
Dsvd_op_gpu.cu.cc181 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/
Dtensor_ops_util.h45 out->flat<dtype>().device(ctx->eigen_device<Device>()) = \ in ZerosLikeTensor()
105 out->flat<dtype>().device(ctx->eigen_device<Device>()) = \ in BinaryAddTensors()

1234567