Home
last modified time | relevance | path

Searched refs:SupportsZeroClamp (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.cc161 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateCode()
171 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateCode()
174 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateCode()
182 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateCode()
189 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_constants.cc153 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateConvolutionConstantCode()
175 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionConstantCode()
184 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConvolutionConstantCode()
205 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionConstantCode()
Dconvolution_transposed.cc225 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConvolutionTransposedCode()
330 if (!src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
339 if (block_size.z == 1 && !src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConvolutionTransposedCode()
357 if (!src_def.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
366 if (block_size.y == 1 && !src_def.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
378 if (!src_def.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
387 if (block_size.x == 1 && !src_def.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
Ddepthwise_conv.cc145 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateDepthwiseConvolutionCode()
175 if (!src_desc.SupportsZeroClamp(Axis::DEPTH)) { in GenerateDepthwiseConvolutionCode()
182 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateDepthwiseConvolutionCode()
192 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateDepthwiseConvolutionCode()
Dconvolution_transposed_4x4.cc209 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
214 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
225 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateConvolutionTransposedCode()
Dconvolution_transposed_3x3.cc186 if (!src_desc.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConvolutionTransposedCode()
191 if (!src_desc.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConvolutionTransposedCode()
203 if (src_desc.HasAxis(axis) && !src_desc.SupportsZeroClamp(axis)) { in GenerateConvolutionTransposedCode()
Dconv_powervr.cc393 if (src_def.HasAxis(axis) && !src_def.SupportsZeroClamp(axis) && in GenerateConv()
647 if (!src_def.SupportsZeroClamp(Axis::DEPTH)) { in GenerateConv()
663 if (!src_def.SupportsZeroClamp(Axis::HEIGHT)) { in GenerateConv()
679 if (!src_def.SupportsZeroClamp(Axis::WIDTH)) { in GenerateConv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.h85 bool SupportsZeroClamp(const Axis& axis) const;
Dtensor_desc.cc919 bool TensorDescriptor::SupportsZeroClamp(const Axis& axis) const { in SupportsZeroClamp() function in tflite::gpu::TensorDescriptor