Home
last modified time | relevance | path

Searched defs:stride_cols (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops.cc66 int* stride_rows, int* stride_cols, int* rate_rows, in ParseSizes()
126 int stride_rows = 0, stride_cols = 0; in Compute() local
225 int stride_rows = 0, stride_cols = 0; in Compute() local
345 int stride_rows = 0, stride_cols = 0; in Compute() local
Ddilation_ops_gpu.cu.cc42 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationKernel()
82 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropInputKernel()
132 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropFilterKernel()
Dconv_ops_using_gemm.cc94 int stride_cols, Padding padding, T3* output_data, in operator ()()
217 int stride_cols, Padding padding, T3* output_data, in operator ()()
520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
Dconv_ops.cc242 int dilation_cols, int stride_rows, int stride_cols, in Run()
282 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
297 int dilation_cols, int stride_rows, int stride_cols, in Run()
461 const int stride_cols = GetTensorDim(params.strides, params.data_format, 'W'); in ComputeConv2DDimension() local
Deigen_backward_cuboid_convolutions_test.cc710 const int stride_cols = 3; in TEST() local
789 const int stride_cols = 3; in TEST() local
866 const int stride_cols = 2; in TEST() local
943 const int stride_cols = 2; in TEST() local
Dconv_ops_fused_image_transform.cc264 int filter_count, int stride_rows, int stride_cols, in operator ()()
820 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute() local
Dconv_grad_filter_ops.cc350 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
Ddeep_conv2d.cc97 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows, in CanUseDeepConv2D()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h83 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder() local
88 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder() local
359 int stride_planes, stride_rows, stride_cols; variable
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc90 const int stride_cols = strides_[3]; in Compute() local
Dextract_image_patches_op.cc81 const int stride_cols = strides_[2]; in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc632 const int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in Conv2DShapeImpl() local
765 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
959 int32 stride_cols; in DepthwiseConv2DNativeShapeImpl() local
1088 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in AvgPoolShape() local
1530 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in MaxPoolShapeImpl() local
1669 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in MaxPoolV2Shape() local
1728 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc489 int32 stride_cols = strides[2]; in CommonFusedConvCalculations() local
1006 int32 stride_cols = strides[2]; in __anon4650176f1702() local
Darray_ops.cc2619 int32 stride_cols = strides[2]; in __anondb9326b23902() local
2716 int32 stride_cols = strides[3]; in __anondb9326b23a02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1859 stride_cols, argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1116 stride_cols = random_int(generator()); in TEST_F() local
2320 stride_cols = random_int(generator()); in TEST_F() local