Searched refs:pad_planes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 224 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
230 pad_planes = std::max<int64>( in launch()
298 const bool planes_odd = (pad_planes % 2 != 0); in launch()
348 CHECK(pad_rows >= 0 && pad_cols >= 0 && pad_planes >= 0) in launch()
350 << pad_planes << ")"; in launch()
380 .set_zero_padding(DimIndex::Z, pad_planes / 2); in launch()
426 {{pad_planes, pad_rows, pad_cols}}, in launch()
|
| D | extract_volume_patches_op.cc | 110 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in Compute() local
113 padding_, &out_planes, &pad_planes)); in Compute()
|
| D | pooling_ops_3d.h | 71 int64 pad_planes; member
|
| D | pooling_ops_3d.cc | 86 &out_plane, &pad_planes)); in Pool3dParameters()
582 const int32 pad_planes = params.pad_planes; in launch() local
611 int p_start = pp * plane_stride - pad_planes; in launch()
815 params.pad_planes, params.pad_rows, params.pad_cols, in launch()
|
| D | pooling_ops_3d_sycl.h | 82 pad_planes_(params.pad_planes), in SYCL3DPoolParams()
|