Searched defs:kernel_w (Results 1 – 9 of 9) sorted by relevance
| /external/ComputeLibrary/src/runtime/heuristics/dwc_native/ |
| D | ClDWCNativeDefaultConfigValhall.cpp | 82 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f32() local
147 const size_t kernel_w = wei_shape[idx_w]; in configure_G78_f16() local
248 const size_t kernel_w = wei_shape[idx_w]; in configure_G77_f16() local
|
| D | ClDWCNativeHeuristicsHelpers.cpp | 42 const size_t kernel_w = weights->tensor_shape()[idx_w]; in use_cl_image_for_weights() local
|
| D | ClDWCNativeDefaultConfigBifrost.cpp | 49 const size_t kernel_w = wei_shape[idx_w]; in configure_f32() local
122 const size_t kernel_w = wei_shape[idx_w]; in configure_f16() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 71 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in MaxPoolForwardNCHW()
114 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
147 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNHWC()
234 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNCHW()
275 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNHWC()
357 const int pooled_width, const int kernel_h, const int kernel_w, in operator ()()
377 const int kernel_h, const int kernel_w, const int stride_h, in operator ()()
428 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in operator ()()
|
| D | pooling_ops_3d_gpu.cu.cc | 35 const int kernel_h, const int kernel_w, const int stride_p, in MaxPoolGradBackwardNoMaskNCDHW()
86 const int kernel_h, const int kernel_w, const int stride_p, in MaxPoolGradBackwardNoMaskNDHWC()
142 const int kernel_w, const int stride_p, const int stride_h, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 86 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | gpu_compatibility.cc | 205 absl::Status CheckKernels(int kernel_h, int kernel_w) { in CheckKernels()
214 absl::Status CheckKernelsAndStrides(int kernel_h, int kernel_w, int strides_h, in CheckKernelsAndStrides()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 533 int64_t kernel_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
603 int64_t kernel_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 885 int64_t kernel_w = tfl_avgpool_op.filter_width(); in matchAndRewrite() local
962 int64_t kernel_w = tfl_maxpool_op.filter_width(); in matchAndRewrite() local
|