Home
last modified time | relevance | path

Searched refs:HasAxis (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dmax_unpooling.cc49 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
62 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
73 std::string src_args = op_def.dst_tensors[0].HasAxis(Axis::DEPTH) in GetMaxUnpoolingKernelCode()
77 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
97 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetMaxUnpoolingKernelCode()
104 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
117 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxUnpoolingKernelCode()
Dpooling.cc55 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetAveragePoolingKernelCode()
58 if (op_def.src_tensors[0].HasAxis(axis)) { in GetAveragePoolingKernelCode()
78 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
106 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
133 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetAveragePoolingKernelCode()
179 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetMaxPoolingKernelCode()
182 if (op_def.src_tensors[0].HasAxis(axis)) { in GetMaxPoolingKernelCode()
198 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxPoolingKernelCode()
238 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxPoolingKernelCode()
246 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetMaxPoolingKernelCode()
[all …]
Ddepthwise_conv.cc93 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
122 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
145 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateDepthwiseConvolutionCode()
160 if (src_desc.HasAxis(axis)) { in GenerateDepthwiseConvolutionCode()
172 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
179 if (op_def.dst_tensors[0].HasAxis(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
186 if (op_def.dst_tensors[0].HasAxis(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
216 if (op_def.dst_tensors[0].HasAxis(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
219 if (op_def.dst_tensors[0].HasAxis(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
222 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
[all …]
Dpadding.cc37 op_def.dst_tensors[0].HasAxis(Axis::BATCH) ? "B" : "0"; in GetPaddingCode()
49 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
66 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
73 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
101 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetPaddingCode()
Dconcat_xy.cc50 if (op_def.src_tensors[0].HasAxis(axis) && axis != Axis::BATCH) { in GetConcatKernelCode()
57 if (op_def.dst_tensors[0].HasAxis(axis)) { in GetConcatKernelCode()
72 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetConcatKernelCode()
79 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetConcatKernelCode()
97 if (op_def.src_tensors[i].HasAxis(Axis::BATCH)) { in GetConcatKernelCode()
Dtranspose.cc30 op_def.dst_tensors[0].HasAxis(Axis::BATCH) ? "B" : "0"; in GetTransposeCode()
33 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
59 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
74 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetTransposeCode()
Dreshapex4.cc30 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
44 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
56 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
Dsplit.cc51 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitCode()
58 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitCode()
90 definition_.src_tensors[0].HasAxis(Axis::BATCH) ? ", B" : ""; in GetSplitChannelsCode()
94 if (definition_.src_tensors[0].HasAxis(Axis::BATCH)) { in GetSplitChannelsCode()
103 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GetSplitChannelsCode()
Dreshape.cc29 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
48 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
64 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetReshapeCode()
Dconvolution_transposed.cc200 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
203 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
206 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
225 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConvolutionTransposedCode()
260 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
283 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
316 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
355 src_def.HasAxis(Axis::DEPTH) ? "src_y_copy" : "src_y"; in GenerateConvolutionTransposedCode()
399 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
420 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
[all …]
Dreduce.cc48 bool HasAxis(const std::vector<Axis>& axis, Axis a) { in HasAxis() function
177 if (op_def.dst_tensors[0].HasAxis(a)) { in GetReduceKernelCode()
178 if (!HasAxis(axis_to_reduce, a)) { in GetReduceKernelCode()
183 const bool channels_reductin = HasAxis(axis_to_reduce, Axis::CHANNELS); in GetReduceKernelCode()
274 if (op_def.dst_tensors[0].HasAxis(a) && !HasAxis(axis_to_reduce, a)) { in GetReduceKernelCode()
282 if (op_def.src_tensors[0].HasAxis(a)) { in GetReduceKernelCode()
328 if (op_def.src_tensors[0].HasAxis(a)) { in GetReduceKernelCode()
409 if (op_def.dst_tensors[0].HasAxis(a)) { in GetReduceKernelCode()
Dstrided_slice.cc109 op_def.dst_tensors[0].HasAxis(Axis::BATCH) ? "B" : "0"; in GetStridedSliceCode()
112 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetStridedSliceCode()
128 if (op_def.src_tensors[0].HasAxis(Axis::BATCH)) { in GetStridedSliceCode()
Dconv_powervr.cc259 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH)) { in GenerateCode()
282 if (definition_.src_tensors[0].HasAxis(Axis::DEPTH) && in BindArguments()
365 if (src_def.HasAxis(Axis::WIDTH)) { in GenerateConv()
368 if (src_def.HasAxis(Axis::HEIGHT)) { in GenerateConv()
371 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
393 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConv()
422 if (src_def.HasAxis(Axis::DEPTH) && !conv_params_.z_kernel_is_1) { in GenerateConv()
485 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
513 src_def.HasAxis(Axis::DEPTH)); in GenerateConv()
586 if (src_def.HasAxis(Axis::DEPTH)) { in GenerateConv()
[all …]
Dconcat_z.cc49 if (op_def.dst_tensors[0].HasAxis(Axis::DEPTH)) { in GetConcatKernelCode()
Ddepthwise_conv_3x3_stride_h2.cc35 if (definition.dst_tensors[0].HasAxis(Axis::BATCH)) { in GetKernelDepthWiseConv3x3StrideH2()
Ddepthwise_conv_3x3.cc72 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GenerateDepthwiseConvCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc98 if (HasAxis(Axis::WIDTH)) { in GetGPUResources()
106 if (HasAxis(Axis::HEIGHT)) { in GetGPUResources()
109 if (HasAxis(Axis::CHANNELS)) { in GetGPUResources()
113 if (HasAxis(Axis::BATCH)) { in GetGPUResources()
116 if (HasAxis(Axis::DEPTH)) { in GetGPUResources()
186 if (HasAxis(Axis::BATCH)) { in PerformSelector()
276 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in GetLinkingContextFromWriteSelector()
516 if (HasAxis(Axis::BATCH) && !IsBatchedWidth()) { in PerformGetWHOffsetSelector()
754 bool TensorDescriptor::HasAxis(Axis axis) const { in HasAxis() function in tflite::gpu::TensorDescriptor
790 if (HasAxis(Axis::BATCH)) { in GetSliceStrideSize()
[all …]
Dgpu_operation.cc63 if (HasAxis(src.layout, Axis::BATCH)) { in IsBatchSupported()
68 if (HasAxis(dst.layout, Axis::BATCH)) { in IsBatchSupported()
Dtensor_desc.h72 bool HasAxis(Axis axis) const;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dshape_test.cc96 EXPECT_TRUE(HasAxis<Layout::OHWDI>(Axis::DEPTH)); in TEST()
97 EXPECT_FALSE(HasAxis<Layout::OHWDI>(Axis::CHANNELS)); in TEST()
Dshape.h90 constexpr bool HasAxis(Axis axis);
93 bool HasAxis(Layout layout, Axis axis);
126 bool has(Axis axis) const { return HasAxis(layout, axis); } in has()
642 constexpr bool HasAxis(Axis axis) {
Dshape.cc122 bool HasAxis(Layout layout, Axis axis) { in HasAxis() function
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_spatial_tensor.cc258 if (descriptor_.HasAxis(Axis::WIDTH)) { in GetGPUResources()
266 if (descriptor_.HasAxis(Axis::HEIGHT)) { in GetGPUResources()
269 if (descriptor_.HasAxis(Axis::CHANNELS)) { in GetGPUResources()
273 if (descriptor_.HasAxis(Axis::BATCH)) { in GetGPUResources()
276 if (descriptor_.HasAxis(Axis::DEPTH)) { in GetGPUResources()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor.cc353 if (descriptor_.HasAxis(Axis::WIDTH)) { in GetGPUResources()
361 if (descriptor_.HasAxis(Axis::HEIGHT)) { in GetGPUResources()
364 if (descriptor_.HasAxis(Axis::CHANNELS)) { in GetGPUResources()
368 if (descriptor_.HasAxis(Axis::BATCH)) { in GetGPUResources()
371 if (descriptor_.HasAxis(Axis::DEPTH)) { in GetGPUResources()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc130 if (op_def.dst_tensors[0].HasAxis(Axis::BATCH)) { in GenerateCode()
161 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateCode()

12