| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops.cc | 67 int* rate_cols, int64* pad_top, int64* pad_left, in ParseSizes() argument
84 *rate_cols = rates[2]; in ParseSizes()
104 filter_cols + (filter_cols - 1) * (*rate_cols - 1); in ParseSizes()
127 int rate_rows = 0, rate_cols = 0; in Compute() local
131 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, in Compute()
151 filter.tensor<T, 3>(), stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
166 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
192 const int w_in = w_beg + w * rate_cols; in operator ()()
226 int rate_rows = 0, rate_cols = 0; in Compute() local
230 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, in Compute()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 43 int rate_cols, int pad_top, int pad_left, T* __restrict__ output_ptr) { in DilationKernel() argument
59 const int w_in = w_beg + w * rate_cols; in DilationKernel()
82 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropInputKernel() argument
104 const int w_in = w_beg + w * rate_cols; in DilationBackpropInputKernel()
132 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropFilterKernel() argument
154 const int w_in = w_beg + w * rate_cols; in DilationBackpropFilterKernel()
183 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
204 rate_cols, pad_top, pad_left, output.data())); in operator ()()
214 int rate_cols, int pad_top, int pad_left, in operator ()()
245 stride_rows, stride_cols, rate_rows, rate_cols, pad_top, pad_left, in operator ()()
[all …]
|
| D | dilation_ops.h | 31 int stride_cols, int rate_rows, int rate_cols, int pad_top,
45 int rate_cols, int pad_top, int pad_left,
59 int rate_cols, int pad_top, int pad_left,
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.h | 30 int stride_cols, int rate_rows, int rate_cols, in operator()
41 stride_rows, rate_cols, rate_rows, padding) in operator()
47 stride_rows, rate_cols, rate_rows, padding) in operator()
|
| D | extract_image_patches_op.cc | 84 const int rate_cols = rates_[2]; in Compute() local
87 const int ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in Compute()
112 ksize_cols, stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
147 int rate_rows, int rate_cols, const Eigen::PaddingType& padding, \
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractImagePatches.pbtxt | 34 Must be: `[1, rate_rows, rate_cols, 1]`. This is the
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 1009 int32 rate_cols = rates[2]; in __anon4650176f1702() local
1035 auto filter_cols_eff = filter_cols + (filter_cols - 1) * (rate_cols - 1); in __anon4650176f1702()
|
| D | array_ops.cc | 2622 int32 rate_cols = rates[2]; in __anondb9326b23902() local
2625 int32 ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in __anondb9326b23902()
|