Home
last modified time | relevance | path

Searched defs:stride_rows (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()
81 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropInputKernel()
131 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropFilterKernel()
Dconv_ops_using_gemm.cc93 int filter_width, int filter_count, int stride_rows, in operator ()()
216 int filter_width, int filter_count, int stride_rows, in operator ()()
519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 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()
460 const int stride_rows = GetTensorDim(params.strides, params.data_format, 'H'); in ComputeConv2DDimension() local
Deigen_backward_cuboid_convolutions_test.cc711 const int stride_rows = 1; in TEST() local
790 const int stride_rows = 1; in TEST() local
865 const int stride_rows = 3; in TEST() local
942 const int stride_rows = 3; in TEST() local
Dconv_ops_fused_image_transform.cc264 int filter_count, int stride_rows, int stride_cols, in operator ()()
819 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute() local
Dconv_grad_filter_ops.cc349 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()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h82 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder() local
87 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder() local
359 int stride_planes, stride_rows, stride_cols; variable
/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc89 const int stride_rows = strides_[2]; in Compute() local
Dextract_image_patches_op.cc80 const int stride_rows = strides_[1]; in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc631 const int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in Conv2DShapeImpl() local
765 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
958 int32 stride_rows; in DepthwiseConv2DNativeShapeImpl() local
1087 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in AvgPoolShape() local
1529 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in MaxPoolShapeImpl() local
1668 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in MaxPoolV2Shape() local
1728 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc488 int32 stride_rows = strides[1]; in CommonFusedConvCalculations() local
1005 int32 stride_rows = strides[1]; in __anon4650176f1702() local
Darray_ops.cc2618 int32 stride_rows = strides[1]; in __anondb9326b23902() local
2715 int32 stride_rows = strides[2]; in __anondb9326b23a02() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1858 stride_rows, argument
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1115 int stride_rows = random_int(generator()), in TEST_F() local
2319 int stride_rows = random_int(generator()), in TEST_F() local