Searched defs:kernel_h (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operation_parser.cc | 28 absl::Status CheckKernels(int kernel_h, int kernel_w) { in CheckKernels()
37 absl::Status CheckKernelsAndStrides(int kernel_h, int kernel_w, int strides_h, in CheckKernelsAndStrides()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 70 const int pooled_height, const int pooled_width, const int kernel_h, 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 ()()
427 const int channels, const int height, const int width, const int kernel_h, 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()
141 const int height, const int width, const int kernel_p, const int kernel_h, in operator ()()
|
| D | avgpooling_op_gpu.cu.cc | 85 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC()
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | cuda_common.h | 9 FILE *kernel_h; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 527 int64_t kernel_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
597 int64_t kernel_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1007 int64_t kernel_h = tfl_avgpool_op.filter_height(); in matchAndRewrite() local
1061 int64_t kernel_h = tfl_maxpool_op.filter_height(); in matchAndRewrite() local
|