Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h76 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder() local
81 int stride_rows = GetTensorDim(strides_, data_format_, '1'); 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.cc43 int output_cols, int stride_rows, 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.cc91 int filter_width, int filter_count, int stride_rows, in operator ()()
214 int filter_width, int filter_count, int stride_rows, in operator ()()
517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local
Dextract_image_patches_op.cc77 const int stride_rows = strides_[1]; in Compute() local
Dextract_volume_patches_op.cc86 const int stride_rows = strides_[2]; 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()
393 const int stride_rows = GetTensorDim(params.strides, params.data_format, 'H'); in ComputeConv2DDimension() local
Dconv_ops_fused_image_transform.cc263 int filter_count, int stride_rows, int stride_cols, in operator ()()
822 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute() local
Deigen_backward_spatial_convolutions_test.cc1811 const int stride_rows = 1; in TEST() local
1890 const int stride_rows = 1; in TEST() local
1965 const int stride_rows = 3; in TEST() local
2042 const int stride_rows = 3; in TEST() local
Dconv_grad_filter_ops.cc519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local
Dconv_grad_input_ops.cc656 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local
Ddeep_conv2d.cc97 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows, in CanUseDeepConv2D()
Dmkl_conv_ops.cc522 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc464 const int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in Conv2DShapeImpl() local
582 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
664 int32 stride_rows; in DepthwiseConv2DNativeShape() local
753 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in AvgPoolShape() local
904 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in MaxPoolShape() local
1006 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in MaxPoolV2Shape() local
1066 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc408 int32 stride_rows = strides[1]; in CommonFusedConvCalculations() local
911 int32 stride_rows = strides[1]; in __anon3e672dd81d02() local
Darray_ops.cc2459 int32 stride_rows = strides[1]; in __anon7c94107b3902() local
2556 int32 stride_rows = strides[2]; in __anon7c94107b3a02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1639 filter_cols, in_depth, out_depth, stride_rows, argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1114 int stride_rows = random_int(generator()), in TEST_F() local
2318 int stride_rows = random_int(generator()), in TEST_F() local