Searched defs:dilation_rows (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops.cc | 386 int out_cols, int out_depth, int dilation_rows, in Run()
428 int dilation_rows, int dilation_cols, Tensor* output, in Run()
441 int out_cols, int out_depth, int dilation_rows, in Run()
610 const int dilation_rows = in ComputeConv2DDimension() local
|
| H A D | conv_grad_filter_ops.cc | 355 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H'); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 386 int dilation_planes, dilation_rows, dilation_cols; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 641 const int32_t dilation_rows = GetTensorDim(dilations, data_format, 'H'); in Conv2DShapeImpl() local
798 int32_t dilation_planes, dilation_rows, dilation_cols; in Conv3DShape() local
1015 int32_t dilation_rows; in DepthwiseConv2DNativeShapeImpl() local
|