Home
last modified time | relevance | path

Searched refs:y_kernel_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.cc145 params_.y_kernel_size * params_.y_dilation_size - 1; in GenerateSrcUpload()
282 kernel_size_y = std::to_string(params_.y_kernel_size); in GenerateCode()
330 "(kz * " + std::to_string(params_.y_kernel_size) + " + ky)"; in GenerateCode()
400 params.y_kernel_size = attr.weights.shape.h; in CreateDepthwiseConvolution2D()
Ddepthwise_conv.h69 return x_kernel_size * y_kernel_size * z_kernel_size; in GetKernelsTotalSize()
86 int y_kernel_size = 1; member