Searched refs:kernel_size_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_powervr.cc | 176 kernel_size_(attr.weights.shape.w, attr.weights.shape.h, 1, 1), in ConvPowerVR()
187 kernel_size_(weights_shape.w, weights_shape.h, 1, 1), in ConvPowerVR()
198 kernel_size_(1, 1, 1, 1), in ConvPowerVR()
206 kernel_size_(1, 1, 1, 1), in ConvPowerVR()
213 kernel_size_(operation.kernel_size_), in ConvPowerVR()
224 kernel_size_(attr.weights.shape.w, attr.weights.shape.h, in ConvPowerVR()
233 std::swap(kernel_size_, operation.kernel_size_); in operator =()
273 RETURN_IF_ERROR(args->SetInt("kernel_size_x", kernel_size_.x)); in BindArguments()
279 RETURN_IF_ERROR(args->SetInt("kernel_size_y", kernel_size_.y)); in BindArguments()
286 RETURN_IF_ERROR(args->SetInt("kernel_size_z", kernel_size_.z)); in BindArguments()
|
| D | conv_powervr.h | 200 int4 kernel_size_; variable
|