Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h77 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder() local
82 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder() local
353 int stride_planes, stride_rows, stride_cols; variable
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.cc44 int stride_cols, int rate_rows, int rate_cols, in DilationKernel()
83 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropInputKernel()
132 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropFilterKernel()
Dconv_ops_using_gemm.cc92 int stride_cols, Padding padding, T3* output_data, in operator ()()
215 int stride_cols, Padding padding, T3* output_data, in operator ()()
518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
Dextract_image_patches_op.cc78 const int stride_cols = strides_[2]; in Compute() local
Dextract_volume_patches_op.cc87 const int stride_cols = strides_[3]; in Compute() local
Dconv_ops.cc175 int dilation_cols, int stride_rows, int stride_cols, in Run()
215 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run()
230 int dilation_cols, int stride_rows, int stride_cols, in Run()
394 const int stride_cols = GetTensorDim(params.strides, params.data_format, 'W'); in ComputeConv2DDimension() local
Dconv_ops_fused_image_transform.cc263 int filter_count, int stride_rows, int stride_cols, in operator ()()
823 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute() local
Deigen_backward_spatial_convolutions_test.cc1810 const int stride_cols = 3; in TEST() local
1889 const int stride_cols = 3; in TEST() local
1966 const int stride_cols = 2; in TEST() local
2043 const int stride_cols = 2; in TEST() local
Dconv_grad_filter_ops.cc520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
Dconv_grad_input_ops.cc657 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()
Dmkl_conv_ops.cc523 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc465 const int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in Conv2DShapeImpl() local
582 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
665 int32 stride_cols; in DepthwiseConv2DNativeShape() local
754 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in AvgPoolShape() local
905 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in MaxPoolShape() local
1007 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in MaxPoolV2Shape() local
1066 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc409 int32 stride_cols = strides[2]; in CommonFusedConvCalculations() local
912 int32 stride_cols = strides[2]; in __anon3e672dd81d02() local
Darray_ops.cc2460 int32 stride_cols = strides[2]; in __anon7c94107b3902() local
2557 int32 stride_cols = strides[3]; in __anon7c94107b3a02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1640 stride_cols, padding, test_input, data_format, argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1115 stride_cols = random_int(generator()); in TEST_F() local
2319 stride_cols = random_int(generator()); in TEST_F() local