| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op_gpu.cu.cc | 47 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,
|
| 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() 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 …]
|
| 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() 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 ()()
|
| D | xsmm_conv2d_test.cc | 47 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()
|
| D | maxpooling_op_gpu.h | 40 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,
|
| D | conv_grad_filter_ops_benchmark_test.cc | 42 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()
|
| D | conv_grad_input_ops_benchmark_test.cc | 41 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()
|
| D | pooling_ops_3d_gpu.h | 39 const int kernel_w, const int stride_p, const int stride_h,
|
| D | avgpooling_op.h | 72 const int kernel_w, const int stride_h,
|
| /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
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/ |
| D | optimize_patterns.td | 52 $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/ |
| D | many_attribute_op.mlir | 6 …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…
|
| D | dynamic_shape.mlir | 7 …ctor = 1 : i32, fused_activation_function = "RELU6", padding = "VALID", stride_h = 2 : i32, stride…
|
| D | import_json.json | 4 …factor = 1 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 0 : i32, stride…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | nn.mlir | 37 // CHECK-NEXT: stride_h: 3,
61 …width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride…
|
| D | transpose_conv_optional.mlir | 50 // CHECK-NEXT: stride_h: 2
77 …%0 = "tfl.transpose_conv"(%arg0, %arg1, %arg2, %cst) {padding = "SAME", stride_h = 2 : i32, stride…
|
| D | depthwise_conv2d.mlir | 68 // CHECK-NEXT: stride_h: 4,
100 …actor = 1 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride…
|
| D | depthwise_conv2d_v2.mlir | 68 // 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/ |
| D | optimize.mlir | 11 …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 …]
|
| D | prepare-quantize.mlir | 65 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/ |
| D | schema_v0.fbs | 115 stride_h:int;
122 stride_h:int;
131 stride_h:int;
|
| D | schema_v1.fbs | 128 stride_h:int;
135 stride_h:int;
144 stride_h:int;
|
| D | schema_v2.fbs | 136 stride_h:int;
143 stride_h:int;
152 stride_h:int;
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_conv_test.cc | 30 int stride_w, int stride_h, in QuantizedTransposeConvOpModel() argument
76 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in QuantizedTransposeConvOpModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv_test.cc | 62 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()
|