Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc89 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); in Compute()
97 functor2(ctx->template eigen_device<Device>(), To, I(0), I(1)); in Compute()
102 functor3(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2)); in Compute()
107 functor4(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
113 functor5(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
119 functor6(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
125 functor7(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
131 functor8(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
138 functor9(ctx->template eigen_device<Device>(), To, I(0), I(1), I(2), in Compute()
147 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()
Dresize_nearest_neighbor_op.cc70 context->eigen_device<Device>(), input_data, st.height_scale, in Compute()
76 context->eigen_device<Device>(), input_data, st.height_scale, in Compute()
84 context->eigen_device<Device>(), input_data, st.height_scale, in Compute()
90 context->eigen_device<Device>(), input_data, st.height_scale, in Compute()
235 context->eigen_device<Device>(), input_data, height_scale, in Compute()
241 context->eigen_device<Device>(), input_data, height_scale, in Compute()
250 context->eigen_device<Device>(), input_data, height_scale, in Compute()
257 context->eigen_device<Device>(), input_data, height_scale, in Compute()
Dcudnn_pooling_gpu.cc54 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
111 context->eigen_device<GPUDevice>(), in Compute()
175 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
180 functor::NHWCToNCHW<GPUDevice, T, 5>()(context->eigen_device<GPUDevice>(), in Compute()
185 context->eigen_device<GPUDevice>(), out_backprop.tensor<T, 5>(), in Compute()
240 context->eigen_device<GPUDevice>(), in Compute()
Dtranspose_op.cc208 return ::tensorflow::DoTranspose(ctx->eigen_device<CPUDevice>(), in, perm, in DoTranspose()
217 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<CPUDevice>(), in, in DoTranspose()
255 return ::tensorflow::DoTranspose(ctx->eigen_device<GPUDevice>(), in, perm, in TF_CALL_ALL_TYPES()
263 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<GPUDevice>(), in, in DoTranspose()
286 return ::tensorflow::DoTranspose(ctx->eigen_device<SYCLDevice>(), in, perm, in DoTranspose()
294 return ::tensorflow::DoConjugateTranspose(ctx->eigen_device<SYCLDevice>(), in, in DoTranspose()
Dfused_batch_norm_op.cc90 const CPUDevice& d = context->eigen_device<CPUDevice>(); in operator ()()
183 const CPUDevice& d = context->eigen_device<CPUDevice>(); in operator ()()
267 f(context->eigen_device<GPUDevice>(), batch_mean->flat<U>()); in operator ()()
268 f(context->eigen_device<GPUDevice>(), batch_var->flat<U>()); in operator ()()
286 context->eigen_device<GPUDevice>(), in operator ()()
331 GPUDevice d = context->eigen_device<GPUDevice>(); in operator ()()
376 context->eigen_device<GPUDevice>(), in operator ()()
426 context->eigen_device<GPUDevice>(), in operator ()()
438 context->eigen_device<GPUDevice>(), in operator ()()
499 context->eigen_device<GPUDevice>(), in operator ()()
[all …]
Dcwise_ops_common.h103 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local
111 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
116 eigen_device, out_flat, in0.template scalar<Tin>(), in Compute()
120 eigen_device, out_flat, in0.template flat<Tin>(), in Compute()
125 eigen_device, out->shaped<Tout, 2>(bcast->result_shape()), in Compute()
132 eigen_device, out->shaped<Tout, 3>(bcast->result_shape()), in Compute()
139 eigen_device, out->shaped<Tout, 4>(bcast->result_shape()), in Compute()
146 eigen_device, out->shaped<Tout, 5>(bcast->result_shape()), in Compute()
180 const Device& d = context->eigen_device<Device>(); in Compute()
209 const Device& eigen_device = ctx->eigen_device<Device>(); in Compute() local
[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()
Dquantize_op.cc155 o.device(ctx->template eigen_device<Device>()) = in Compute()
179 ctx->template eigen_device<Device>(), input, min_range, max_range, in Compute()
199 o.device(ctx->template eigen_device<Device>()) = in Compute()
206 o.device(ctx->template eigen_device<Device>()) = in Compute()
Dpooling_ops_common.cc186 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
277 context->eigen_device<Device>(), in Compute()
358 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
366 functor::NHWCToNCHW<GPUDevice, T, 4>()(context->eigen_device<Device>(), in Compute()
371 context->eigen_device<Device>(), out_backprop.tensor<T, 4>(), in Compute()
429 context->eigen_device<Device>(), in Compute()
Dbias_op.cc142 const Device& d = context->eigen_device<Device>(); in Compute()
151 const Device& d = context->eigen_device<Device>(); in Compute()
160 const Device& d = context->eigen_device<Device>(); in Compute()
198 functor(ctx->eigen_device<Device>(), input.tensor<T, Dims>(), bias.vec<T>(), in Compute()
283 output->template flat<T>().device(context->eigen_device<Device>()) = in Compute()
297 redux(context->eigen_device<Device>(), two_dims, output_backprop, in Compute()
366 BiasGPU<T>::compute(context->template eigen_device<Device>(), in Compute()
494 BiasGradGPU<T>::compute(context->template eigen_device<Device>(), in ComputeWithCustomKernel()
Dsvd_op_gpu.cu.cc147 const GPUDevice& d = context->eigen_device<GPUDevice>(); in RunSVD()
196 const GPUDevice& d = context->eigen_device<GPUDevice>(); in RunSVD()
210 auto device = context->eigen_device<GPUDevice>(); in RunSVD()
254 auto device = context->eigen_device<GPUDevice>(); in PerformSVD_MgeqN()
282 const GPUDevice& d = context->eigen_device<GPUDevice>(); in PerformSVD_MlessN()
Dfake_quant_ops.cc79 functor(context->eigen_device<Device>(), input.flat<float>(), min_, max_, in Operate()
128 functor(context->eigen_device<Device>(), gradient.flat<float>(), in OperateNoTemplate()
200 functor(context->eigen_device<Device>(), input.flat<float>(), in Compute()
251 functor(context->eigen_device<Device>(), gradient.flat<float>(), in Compute()
336 functor(context->eigen_device<Device>(), input.flat_inner_dims<float, 2>(), in Compute()
396 context->eigen_device<Device>(), gradient.flat_inner_dims<float, 2>(), in Compute()
Ddepthtospace_op.cc121 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_, in Compute()
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_, in Compute()
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
Dspacetodepth_op.cc123 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_, in Compute()
128 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_, in Compute()
139 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput); in Compute()
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()
Ddiag_op_gpu.cu.cc61 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()()
100 const GPUDevice& device = context->eigen_device<GPUDevice>(); in operator ()()
Dscatter_op.cc131 const Index bad_i = functor(c, c->template eigen_device<Device>(), in DoCompute()
143 const Index bad_i = functor(c, c->template eigen_device<Device>(), in DoCompute()
219 const Index bad_i = functor(c, c->template eigen_device<SYCLDevice>(), in DoCompute()
231 const Index bad_i = functor(c, c->template eigen_device<SYCLDevice>(), in DoCompute()
Dtensor_array.cc35 add_functor(ctx->template eigen_device<Device>(), sum->flat<T>(), \
60 set_zero_functor(ctx->template eigen_device<Device>(), value->flat<T>()); \
Ddense_update_functor.cc77 copy_functor_(context->eigen_device<CPUDevice>(), tensor->flat<T>(), \
112 copy_functor_(context->eigen_device<GPUDevice>(), tensor->flat<T>(), \
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()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base_test.cc33 Eigen::ThreadPoolDevice eigen_device(&wrapper, pool.NumThreads()); in TEST() local
35 dbase.set_eigen_cpu_device(&eigen_device); in TEST()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsegmentation_ops.cc57 TensorRangeFunctor<Device>()(ctx->eigen_device<Device>(), in Compute()
60 rank.device(ctx->eigen_device<Device>()) = rank.constant(OutputType(0)); in Compute()
116 FindRootFunctor<CPUDevice, T>()(ctx->eigen_device<CPUDevice>(), output, in operator ()()
/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()

123456