Home
last modified time | relevance | path

Searched refs:stride_h (Results 1 – 25 of 61) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op_gpu.cu.cc47 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in DEFINE_GPU_KERNELS()
56 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in DEFINE_GPU_KERNELS()
57 const int phend = min(h / stride_h + 1, pooled_height); in DEFINE_GPU_KERNELS()
66 int hstart = ph * stride_h - pad_t; in DEFINE_GPU_KERNELS()
86 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC() argument
95 channels, pooled_height, pooled_width, kernel_h, kernel_w, stride_h, in RunAvePoolBackwardNHWC()
105 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
111 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
117 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
Dmaxpooling_op_gpu.cu.cc71 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in MaxPoolForwardNCHW() argument
79 int hstart = ph * stride_h - 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() argument
124 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC() argument
157 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolForwardNHWC()
234 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNCHW() argument
243 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCHW()
275 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNHWC() argument
285 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNHWC()
[all …]
Dpooling_ops_3d_gpu.cu.cc36 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNCDHW() argument
47 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC() argument
99 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNDHWC()
142 const int kernel_w, const int stride_p, const int stride_h, in operator ()() argument
153 height, width, kernel_p, kernel_h, kernel_w, stride_p, stride_h, in operator ()()
160 height, width, kernel_p, kernel_h, kernel_w, stride_p, stride_h, in operator ()()
Dxsmm_conv2d_test.cc47 int stride_h; member
131 int stride_h = param->stride_h; in naive_conv_fp() local
146 ij = oj * stride_h - pad_h; in naive_conv_fp()
201 int stride_h = stride; in TEST_F() local
212 int ofh = (ifh + 2 * pad_h - kh) / stride_h + 1; in TEST_F()
263 naive_param.stride_h = stride_h; in TEST_F()
Dmaxpooling_op_gpu.h40 const int kernel_w, const int stride_h, const int stride_w,
50 const int kernel_w, const int stride_h, const int stride_w,
77 const int kernel_h, const int kernel_w, const int stride_h,
Dconv_grad_filter_ops_benchmark_test.cc42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter() argument
55 params.strides = {1, stride_h, stride_w, 1}; in Conv2DBackpropFilter()
87 .Attr("strides", {1, stride_h, stride_w, 1}) in Conv2DBackpropFilter()
Dconv_grad_input_ops_benchmark_test.cc41 int out_depth, int stride_h, int stride_w, in Conv2DBackpropInput() argument
54 params.strides = {1, stride_h, stride_w, 1}; in Conv2DBackpropInput()
86 .Attr("strides", {1, stride_h, stride_w, 1}) in Conv2DBackpropInput()
Dpooling_ops_3d_gpu.h39 const int kernel_w, const int stride_p, const int stride_h,
Davgpooling_op.h72 const int kernel_w, const int stride_h,
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dops_defs.py57 def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, argument
62 strides=[1, stride_w, stride_h, 1],
176 def _composite_max_pool(input_, stride_w, stride_h, filter_width, filter_height, argument
179 strides = [1, stride_w, stride_h, 1]
189 stride_h = op.get_attr('stride_h')
196 strides=[1, stride_w, stride_h, 1],
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_patterns.td52 $w_factor, TFL_AF_None, $padding, $stride_h, $stride_w)),
54 $padding, $stride_h, $stride_w),
58 $w_factor, TFL_AF_None, $padding, $stride_h, $stride_w,
61 ActFnAttr, $padding, $stride_h, $stride_w, $multiplier),
68 $filter_width, $padding, $stride_h, $stride_w, TFL_AF_None)),
70 $stride_h, $stride_w, ActFnAttr),
73 (ActFnOp (TFL_MaxPool2DOp:$pool_out $input, $padding, $stride_w, $stride_h,
75 (TFL_MaxPool2DOp $input, $padding, $stride_w, $stride_h,
102 TFL_AF_None, $padding, $stride_h, $stride_w),
107 $h_factor, $w_factor, $act_fn, $padding, $stride_h, $stride_w),
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dmany_attribute_op.mlir6 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
7 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
Ddynamic_shape.mlir7 …ctor = 1 : i32, fused_activation_function = "RELU6", padding = "VALID", stride_h = 2 : i32, stride…
Dimport_json.json4 …factor = 1 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 0 : i32, stride…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dnn.mlir37 // CHECK-NEXT: stride_h: 3,
61 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
Dtranspose_conv_optional.mlir50 // CHECK-NEXT: stride_h: 2
77 …%0 = "tfl.transpose_conv"(%arg0, %arg1, %arg2, %cst) {padding = "SAME", stride_h = 2 : i32, stride…
Ddepthwise_conv2d.mlir68 // CHECK-NEXT: stride_h: 4,
100 …actor = 1 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride…
Ddepthwise_conv2d_v2.mlir68 // CHECK-NEXT: stride_h: 4,
102 …actor = 2 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Doptimize.mlir11 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
15 …factor = 3 : i32, fused_activation_function = "RELU", padding = "SAME", stride_h = 4 : i32, stride…
21 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
25 …actor = 3 : i32, fused_activation_function = "RELU6", padding = "SAME", stride_h = 4 : i32, stride…
31 …width = 3 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride…
35 …width = 3 : i32, fused_activation_function = "RELU", padding = "VALID", stride_h = 2 : i32, stride…
41 …width = 3 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride…
45 …3 : i32, fused_activation_function = "RELU_N1_TO_1", padding = "VALID", stride_h = 2 : i32, stride…
53 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
65 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
[all …]
Dprepare-quantize.mlir65 fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32}
85 fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32}
105 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
149 …actor = 1 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride…
169 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
448 …width = 3 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride…
452 …width = 3 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride…
466 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
469 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
581 …factor = 3 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride…
[all …]
/external/tensorflow/tensorflow/lite/schema/
Dschema_v0.fbs115 stride_h:int;
122 stride_h:int;
131 stride_h:int;
Dschema_v2.fbs136 stride_h:int;
143 stride_h:int;
152 stride_h:int;
Dschema_v1.fbs128 stride_h:int;
135 stride_h:int;
144 stride_h:int;
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc30 int stride_w, int stride_h, in QuantizedTransposeConvOpModel() argument
76 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in QuantizedTransposeConvOpModel()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc62 Padding padding, int stride_w, int stride_h, in BaseTransposeConvOpModel() argument
80 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in BaseTransposeConvOpModel()
579 int stride_w, int stride_h, TestType test_type, in BaseTransposeConvBiasOpModel() argument
626 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in BaseTransposeConvBiasOpModel()

123