| /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()
58 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in DEFINE_GPU_KERNELS()
59 const int pwend = min(w / stride_w + 1, pooled_width); in DEFINE_GPU_KERNELS()
67 int wstart = pw * stride_w - pad_l; in DEFINE_GPU_KERNELS()
87 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC() argument
96 stride_w, pad_t, pad_t, bottom_diff)); 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
80 int wstart = pw * stride_w - pad_l; 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
125 int wstart = pw * stride_w - 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() argument
155 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolForwardNHWC()
235 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
244 int wstart = pw * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNCHW()
276 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC() argument
283 int wstart = (n % pooled_width) * stride_w - pad_l; 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
48 int wstart = pw * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC() argument
95 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNDHWC()
143 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()() argument
154 stride_w, pad_p, pad_t, pad_l, top_diff, bottom_diff)); in operator ()()
161 stride_w, pad_p, pad_t, pad_l, top_diff, bottom_diff)); in operator ()()
|
| D | xsmm_conv2d_test.cc | 48 int stride_w; member
132 int stride_w = param->stride_w; in naive_conv_fp() local
148 ii = oi * stride_w - pad_w; in naive_conv_fp()
200 int stride_w = stride; in TEST_F() local
213 int ofw = (ifw + 2 * pad_w - kw) / stride_w + 1; in TEST_F()
264 naive_param.stride_w = stride_w; 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,
78 const int stride_w, const int pad_t, const int pad_l,
|
| 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 | 40 const int stride_w, const int pad_p, const int pad_t,
|
| D | avgpooling_op.h | 73 const int stride_w, const int pad_t,
|
| /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]
188 stride_w = op.get_attr('stride_w')
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 …ed_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32} : (ten…
7 …ed_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32} : (ten…
|
| D | dynamic_shape.mlir | 7 …d_activation_function = "RELU6", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
|
| D | import_json.json | 4 …sed_activation_function = "NONE", padding = "SAME", stride_h = 0 : i32, stride_w = 0 : i32} : (ten…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | nn.mlir | 36 // CHECK-NEXT: stride_w: 1,
61 …ed_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32} : (ten…
|
| D | transpose_conv_optional.mlir | 49 // CHECK-NEXT: stride_w: 2,
77 …conv"(%arg0, %arg1, %arg2, %cst) {padding = "SAME", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
|
| D | depthwise_conv2d.mlir | 67 // CHECK-NEXT: stride_w: 5,
100 …ed_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
|
| D | depthwise_conv2d_v2.mlir | 67 // CHECK-NEXT: stride_w: 5,
102 …ed_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | optimize.mlir | 11 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
15 …sed_activation_function = "RELU", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
21 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
25 …ed_activation_function = "RELU6", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
31 …ed_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
35 …ed_activation_function = "RELU", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
41 …ed_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
45 …ation_function = "RELU_N1_TO_1", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
53 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
65 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
[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 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
149 …ed_activation_function = "NONE", padding = "VALID", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
169 … i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32
448 …ed_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
452 …ed_activation_function = "NONE", padding = "VALID", stride_h = 2 : i32, stride_w = 2 : i32} : (ten…
466 … i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32
469 …: i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32
581 …sed_activation_function = "NONE", padding = "SAME", stride_h = 4 : i32, stride_w = 5 : i32} : (ten…
[all …]
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v0.fbs | 114 stride_w:int;
121 stride_w:int;
130 stride_w:int;
|
| D | schema_v2.fbs | 135 stride_w:int;
142 stride_w:int;
151 stride_w:int;
|
| D | schema_v1.fbs | 127 stride_w:int;
134 stride_w:int;
143 stride_w: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()
|