| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 57 def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, argument
176 def _composite_max_pool(input_, stride_w, stride_h, filter_width, filter_height, argument
|
| /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()
115 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, 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()
358 const int stride_h, const int stride_w, const int pad_t, const int pad_l, 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 | 36 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC()
142 const int kernel_w, const int stride_p, const int stride_h, in operator ()()
|
| D | xsmm_conv2d_test.cc | 47 int stride_h; member
131 int stride_h = param->stride_h; in naive_conv_fp() local
201 int stride_h = stride; in TEST_F() local
|
| D | avgpooling_op_gpu.cu.cc | 86 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC()
|
| D | conv_grad_input_ops.h | 80 const int stride_h, const int stride_w, T* __restrict im_data) { in Col2im()
|
| D | conv_grad_filter_ops_benchmark_test.cc | 42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter()
|
| D | conv_grad_input_ops_benchmark_test.cc | 41 int out_depth, int stride_h, int stride_w, in Conv2DBackpropInput()
|
| D | conv_grad_filter_ops.cc | 73 const int stride_h, const int stride_w, T* col_data) { in Im2col()
272 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
|
| D | conv_grad_ops_3d.cc | 78 const int pad_r, const int stride_p, const int stride_h, in Col2im()
127 const int pad_r, const int stride_p, const int stride_h, in Im2col()
|
| D | conv_ops.cc | 373 const int64 stride_h = GetTensorDim(strides, data_format, 'H'); in InitConv2DParameters() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv_test.cc | 62 Padding padding, int stride_w, int stride_h, in BaseTransposeConvOpModel()
579 int stride_w, int stride_h, TestType test_type, in BaseTransposeConvBiasOpModel()
|
| D | pooling_test.cc | 36 Padding padding = Padding_VALID, int stride_w = 2, int stride_h = 2, in BasePoolingOpModel()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_conv_test.cc | 30 int stride_w, int stride_h, in QuantizedTransposeConvOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 516 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
586 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
875 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
966 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 1015 int64_t stride_h = tfl_avgpool_op.stride_h(); in matchAndRewrite() local
1069 int64_t stride_h = tfl_maxpool_op.stride_h(); in matchAndRewrite() local
1131 int64_t stride_h = tfl_conv2d_op.stride_h(); in matchAndRewrite() local
1221 int64_t stride_h = tfl_conv_op.stride_h(); in matchAndRewrite() local
1347 int64_t stride_h = tfl_conv2d_op.stride_h(); in matchAndRewrite() local
|
| D | legalize_common.cc | 2820 int64_t stride_h = strides_attr[1].cast<IntegerAttr>().getInt(); in convertTFConv2DCommon() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5665 const int stride_h, const int stride_w, T* im_data) { in Col2im()
|