| /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 | 44 int stride_cols, int rate_rows, int rate_cols, in DilationKernel() argument
61 const int w_in = w_beg + w * rate_cols; in DilationKernel()
84 int rate_cols, int pad_top, int pad_left, T* in_backprop_ptr) { in DilationBackpropInputKernel() argument
105 const int w_in = w_beg + w * rate_cols; in DilationBackpropInputKernel()
133 int rate_cols, int pad_top, int pad_left, T* filter_backprop_ptr) { 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 ()()
203 output_cols, stride_rows, stride_cols, rate_rows, rate_cols, pad_top, 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 | 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 | 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,
|
| D | extract_image_patches_op.cc | 81 const int rate_cols = rates_[2]; in Compute() local
84 const int ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in Compute()
109 ksize_cols, stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
143 int rate_rows, int rate_cols, const Eigen::PaddingType& padding, \
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractImagePatches.pbtxt | 34 1-D of length 4. Must be: `[1, rate_rows, rate_cols, 1]`. This is the
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 915 int32 rate_cols = rates[2]; in __anon3e672dd81d02() local
941 auto filter_cols_eff = filter_cols + (filter_cols - 1) * (rate_cols - 1); in __anon3e672dd81d02()
|
| D | array_ops.cc | 2463 int32 rate_cols = rates[2]; in __anon7c94107b3902() local
2466 int32 ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in __anon7c94107b3902()
|