| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 53 def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, argument
172 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()
235 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW()
276 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC()
358 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()()
378 const int stride_w, const int pad_t, const int pad_l, T* top_data, 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()
143 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()()
|
| D | xsmm_conv2d_test.cc | 48 int stride_w; member
132 int stride_w = param->stride_w; in naive_conv_fp() local
200 int stride_w = stride; in TEST_F() local
|
| D | avgpooling_op_gpu.cu.cc | 87 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC()
|
| D | conv_grad_input_ops.h | 81 const int stride_h, const int stride_w, T* __restrict im_data) { in Col2im()
|
| 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_benchmark_test.cc | 42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter()
|
| D | conv_grad_filter_ops.cc | 77 const int stride_h, const int stride_w, T* col_data) { in Im2col()
277 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DBackpropFilterOp() local
|
| D | depthwise_conv_op.cc | 287 const int64_t stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp() local
|
| D | depthwise_conv_grad_op.cc | 569 const int64_t stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp() local
1072 const int64_t stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropFilterOp() local
|
| D | conv_grad_ops_3d.cc | 83 const int stride_w, T* im_data) { in Col2im()
132 const int stride_w, T* col_data) { in Im2col()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| D | tf_to_xla_attribute_utils.cc | 95 const int stride_w = strides[2].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
171 const int stride_w = strides[2].cast<IntegerAttr>().getInt(); in CalculatePaddingAndPadIfNeeded() local
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEStackLayerKernel.cpp | 131 …const int stride_w = _output->info()->num_dimensions() >= 3 ? _output->info()->strides_in_bytes()[… in run() local
|
| D | NEFFTDigitReverseKernel.cpp | 216 const size_t stride_w = _input->info()->strides_in_bytes()[3]; in digit_reverse_kernel_axis_1() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | optional_input.json | 71 "stride_w": 1, number
|
| D | import_json.json | 76 "stride_w": 1, number
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv_test.cc | 62 Padding padding, int stride_w, int stride_h, in BaseTransposeConvOpModel()
588 int stride_w, int stride_h, TestType test_type, in BaseTransposeConvBiasOpModel()
|
| D | pooling3d_test.cc | 42 int stride_d = 2, int stride_h = 2, int stride_w = 2) { in BasePoolingOpModel()
|
| 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 | 32 int stride_w, int stride_h, in QuantizedTransposeConvOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | replace_cast_hacks_with_tf_xla_ops.cc | 60 const int stride_w = strides[2].cast<IntegerAttr>().getInt(); in PrepareXlaConvParams() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 522 int64_t stride_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
592 int64_t stride_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
879 int64_t stride_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
972 int64_t stride_w = tmpAttr[2].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| D | legalize_tfl.cc | 893 int64_t stride_w = tfl_avgpool_op.stride_w(); in matchAndRewrite() local
970 int64_t stride_w = tfl_maxpool_op.stride_w(); in matchAndRewrite() local
1033 int64_t stride_w = tfl_conv2d_op.stride_w(); in matchAndRewrite() local
1125 int64_t stride_w = tfl_conv_op.stride_w(); in matchAndRewrite() local
1265 int64_t stride_w = tfl_conv2d_op.stride_w(); in matchAndRewrite() local
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuScaleKernel.cpp | 491 const int32_t stride_w = src->info()->strides_in_bytes()[idx_width]; in scale_bilinear_qasymm() local
|