Home
last modified time | relevance | path

Searched refs:CPUDevice (Results 1 – 25 of 296) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops_cpu.h24 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
33 struct Add2Functor<CPUDevice, T> {
34 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
37 Add2EigenImpl<CPUDevice, T>::Compute(d, out, in1, in2);
41 struct Add3Functor<CPUDevice, T> {
42 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
46 Add3EigenImpl<CPUDevice, T>::Compute(d, out, in1, in2, in3);
50 struct Add4Functor<CPUDevice, T> {
51 void operator()(const CPUDevice& d, typename TTypes<T>::Flat out,
56 Add4EigenImpl<CPUDevice, T>::Compute(d, out, in1, in2, in3, in4);
[all …]
Dtranspose_functor_cpu.cc29 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
35 void TransposeSimple(const CPUDevice& device, const Tensor& in, in TransposeSimple()
71 struct Transpose<CPUDevice, T, conjugate> {
72 static void run(const CPUDevice& d, const Tensor& in, in run()
76 internal::TransposeUsingEigen<CPUDevice, T, 2>(d, in, perm, conjugate, in run()
80 internal::TransposeUsingEigen<CPUDevice, T, 3>(d, in, perm, conjugate, in run()
84 internal::TransposeUsingEigen<CPUDevice, T, 4>(d, in, perm, conjugate, in run()
88 internal::TransposeUsingEigen<CPUDevice, T, 5>(d, in, perm, conjugate, in run()
92 internal::TransposeUsingEigen<CPUDevice, T, 6>(d, in, perm, conjugate, in run()
96 internal::TransposeUsingEigen<CPUDevice, T, 7>(d, in, perm, conjugate, in run()
[all …]
Dcwise_op_clip.cc20 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
91 struct UnaryClipOp<CPUDevice, T> {
92 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
110 struct BinaryRightClipOp<CPUDevice, T> {
111 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
130 struct BinaryLeftClipOp<CPUDevice, T> {
131 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
142 struct TernaryClipOp<CPUDevice, T> {
143 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat& in0_flat, in operator ()()
152 template struct UnaryClipOp<CPUDevice, T>; \
[all …]
Ddense_update_functor.h27 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
42 struct DenseUpdate<CPUDevice, T, ADD> {
43 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
50 struct DenseUpdate<CPUDevice, T, SUB> {
51 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
58 struct DenseUpdate<CPUDevice, T, ASSIGN> {
59 void operator()(const CPUDevice& d, typename TTypes<T>::Flat params,
72 Status VariantCopyFn<CPUDevice>(OpKernelContext* context, const Tensor& from,
Dconv_3d.h40 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
43 struct CuboidConvolution<CPUDevice, T> {
44 void operator()(const CPUDevice& d, typename TTypes<T, 5>::Tensor output,
55 struct CuboidConvolutionBackwardInput<CPUDevice, T> {
56 void operator()(const CPUDevice& d,
72 struct CuboidConvolutionBackwardFilter<CPUDevice, T> {
73 void operator()(const CPUDevice& d,
Dl2loss_op.cc29 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
32 class L2LossOp<CPUDevice, T> : public OpKernel {
44 const CPUDevice& d = context->eigen_device<CPUDevice>(); in Compute()
53 L2LossOp<CPUDevice, T>);
Drelu_op.cc31 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
37 ReluOp<CPUDevice, type>); \
40 ReluGradOp<CPUDevice, type>); \
43 Relu6Op<CPUDevice, type>); \
46 Relu6GradOp<CPUDevice, type>) \
49 LeakyReluOp<CPUDevice, type>); \
52 LeakyReluGradOp<CPUDevice, type>);
60 EluOp<CPUDevice, type>); \
63 EluGradOp<CPUDevice, type>); \
66 SeluOp<CPUDevice, type>); \
[all …]
Dquantize_and_dequantize_op.cc34 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
347 struct QuantizeAndDequantizeOneScaleFunctor<CPUDevice, T> {
348 void operator()(const CPUDevice& d, typename TTypes<T>::ConstVec input, in operator ()()
353 QuantizeAndDequantizeOneScaleImpl<CPUDevice, T>::Compute( in operator ()()
360 struct QuantizeAndDequantizePerChannelFunctor<CPUDevice, T> {
361 void operator()(const CPUDevice& d, typename TTypes<T, 3>::ConstTensor input, in operator ()()
366 QuantizeAndDequantizePerChannelImpl<CPUDevice, T>::Compute( in operator ()()
373 struct QuantizeAndDequantizeOneScaleGradientFunctor<CPUDevice, T> {
374 void operator()(const CPUDevice& d, typename TTypes<T>::ConstFlat gradient, in operator ()()
381 QuantizeAndDequantizeOneScaleGradientImpl<CPUDevice, T>::Compute( in operator ()()
[all …]
Dinplace_ops.cc27 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
43 Status DoParallelConcat(const CPUDevice& d, const Tensor& value, int32 loc, in DoParallelConcat()
49 return DoParallelConcatUpdate<CPUDevice, type>(d, value, loc, output); in DoParallelConcat()
138 ParallelConcatUpdate<CPUDevice>);
146 ParallelConcatStart<CPUDevice, type>)
195 ParallelConcatUpdate<CPUDevice>);
244 void DoInplaceOp(const CPUDevice& d, InplaceOpType op, const Tensor& i, in DoInplaceOp()
267 void DoInplaceStringUpdateOp(const CPUDevice& d, const Tensor& i, in DoInplaceStringUpdateOp()
280 Status DoInplace(const CPUDevice& device, InplaceOpType op, const Tensor& i, in DoInplace()
354 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
[all …]
Dsegment_reduction_ops_impl_5.cc35 SparseSegmentReductionSumOp<CPUDevice, type, index_type, \
43 SparseSegmentReductionSumWithNumSegmentsOp<CPUDevice, type, index_type, \
55 SparseSegmentReductionMeanOp<CPUDevice, type, index_type, \
63 SparseSegmentReductionMeanWithNumSegmentsOp<CPUDevice, type, index_type, \
77 SparseSegmentReductionSqrtNOp<CPUDevice, type, index_type, \
86 CPUDevice, type, index_type, segment_ids_type>);
Ddequantize_op.cc39 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
234 DequantizeOp<CPUDevice, quint8, float>);
239 DequantizeOp<CPUDevice, qint8, float>);
244 DequantizeOp<CPUDevice, quint16, float>);
249 DequantizeOp<CPUDevice, qint16, float>);
254 DequantizeOp<CPUDevice, qint32, float>);
260 DequantizeOp<CPUDevice, quint8, bfloat16>);
265 DequantizeOp<CPUDevice, qint8, bfloat16>);
270 DequantizeOp<CPUDevice, quint16, bfloat16>);
275 DequantizeOp<CPUDevice, qint16, bfloat16>);
[all …]
Dsoftmax_op.cc30 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
44 struct SoftmaxFunctor<CPUDevice, T> : SoftmaxFunctorBase<CPUDevice, T> {};
77 SoftmaxOp<CPUDevice, T>);
84 SoftmaxOp<CPUDevice, T>);
Ddense_update_functor.cc28 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
34 struct DenseUpdate<CPUDevice, string, ASSIGN> {
35 void operator()(const CPUDevice& d, typename TTypes<tstring>::Flat params, in operator ()()
76 functor::DenseUpdate<CPUDevice, T, ASSIGN> copy_functor_; \
77 copy_functor_(context->eigen_device<CPUDevice>(), tensor->flat<T>(), \
106 INSTANTIATE_GET_VARIANT_COPY_FN(CPUDevice, TF_CALL_ALL_TYPES, CPU_DENSE_COPY);
Dhistogram_op.cc29 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
35 struct HistogramFixedWidthFunctor<CPUDevice, T, Tout> {
40 const CPUDevice& d = context->eigen_device<CPUDevice>(); in Compute()
126 HistogramFixedWidthOp<CPUDevice, type, int32>) \
131 HistogramFixedWidthOp<CPUDevice, type, int64>)
Dtile_functor_cpu_double.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, double, int32>;
26 template struct Tile<CPUDevice, double, int64>;
Dtile_functor_cpu_float.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, float, int32>;
26 template struct Tile<CPUDevice, float, int64>;
Dtile_functor_cpu_bool.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, bool, int32>;
26 template struct Tile<CPUDevice, bool, int64>;
Dtile_functor_cpu_int64.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, int64, int32>;
26 template struct Tile<CPUDevice, int64, int64>;
Dtile_functor_cpu_tstring.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, tstring, int32>;
26 template struct Tile<CPUDevice, tstring, int64>;
Dtile_functor_cpu_uint32.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, uint32, int32>;
26 template struct Tile<CPUDevice, uint32, int64>;
Dtile_functor_cpu_uint64.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, uint64, int32>;
26 template struct Tile<CPUDevice, uint64, int64>;
Dtile_functor_cpu_variant.cc24 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
26 template struct Tile<CPUDevice, Variant, int32>;
27 template struct Tile<CPUDevice, Variant, int64>;
Dtile_functor_cpu_uint8.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
25 template struct Tile<CPUDevice, uint8, int32>;
26 template struct Tile<CPUDevice, uint8, int64>;
/external/tensorflow/tensorflow/core/framework/
Dregister_types_traits.h21 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
38 struct proxy_type_pod<CPUDevice, 16> {
42 struct proxy_type_pod<CPUDevice, 8> {
46 struct proxy_type_pod<CPUDevice, 4> {
50 struct proxy_type_pod<CPUDevice, 2> {
54 struct proxy_type_pod<CPUDevice, 1> {
/external/tensorflow/tensorflow/core/kernels/image/
Dimage_ops.cc34 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
36 template struct FillProjectiveTransform<CPUDevice, uint8>;
37 template struct FillProjectiveTransform<CPUDevice, int32>;
38 template struct FillProjectiveTransform<CPUDevice, int64>;
39 template struct FillProjectiveTransform<CPUDevice, Eigen::half>;
40 template struct FillProjectiveTransform<CPUDevice, float>;
41 template struct FillProjectiveTransform<CPUDevice, double>;
45 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
157 ImageProjectiveTransformV2<CPUDevice, TYPE>)
180 ImageProjectiveTransformV3<CPUDevice, TYPE>)

12345678910>>...12