Home
last modified time | relevance | path

Searched refs:stride_correction (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpooling.cc28 bool stride_correction, in GetAveragePoolingKernelCode() argument
92 if (stride_correction) { in GetAveragePoolingKernelCode()
146 bool stride_correction, bool output_indices, in GetMaxPoolingKernelCode() argument
214 if (stride_correction) { in GetMaxPoolingKernelCode()
297 const bool stride_correction = in CreatePooling() local
300 op.code_ = GetAveragePoolingKernelCode(definition, stride_correction, &op); in CreatePooling()
302 op.code_ = GetMaxPoolingKernelCode(definition, stride_correction, in CreatePooling()
321 const bool stride_correction = in CreatePooling() local
324 op.code_ = GetAveragePoolingKernelCode(definition, stride_correction, &op); in CreatePooling()
326 op.code_ = GetMaxPoolingKernelCode(definition, stride_correction, in CreatePooling()
Ddepthwise_conv.cc70 const OperationDef& op_def, bool stride_correction, int channel_multiplier, in GenerateDepthwiseConvolutionCode() argument
106 if (stride_correction) { in GenerateDepthwiseConvolutionCode()
253 const bool stride_correction = in CreateDepthwiseConvolution2D() local
255 op.code_ = GenerateDepthwiseConvolutionCode(definition, stride_correction, in CreateDepthwiseConvolution2D()
282 const bool stride_correction = in CreateDepthwiseConvolution2DDynamicWeights() local
284 op.code_ = GenerateDepthwiseConvolutionCode(definition, stride_correction, 1, in CreateDepthwiseConvolution2DDynamicWeights()
321 const bool stride_correction = in CreateDepthwiseConvolution3D() local
323 op.code_ = GenerateDepthwiseConvolutionCode(definition, stride_correction, in CreateDepthwiseConvolution3D()
Dconv_constants.cc99 bool stride_correction, in GenerateConvolutionConstantCode() argument
128 if (stride_correction) { in GenerateConvolutionConstantCode()
282 const bool stride_correction = in CreateConvConstants() local
286 definition, attr.weights.shape, stride_correction, use_dot_conv, &op); in CreateConvConstants()
Dconv_metal.cc160 bool stride_correction) { in GenerateConvolution() argument
286 if (stride_correction) { in GenerateConvolution()
996 const bool stride_correction = in CreateConvolutionMetal() local
998 desc.code_ = GenerateConvolution(params, definition, stride_correction); in CreateConvolutionMetal()
Dconv_powervr.cc247 const bool stride_correction = in GenerateCode() local
249 code_ = GenerateConv(gpu_info, definition_, stride_correction, conv_params_); in GenerateCode()
339 bool stride_correction, in GenerateConv() argument
549 if (stride_correction) { in GenerateConv()
Dconv_powervr.h195 bool stride_correction,