Searched refs:pad_planes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 113 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in Compute() local
116 padding_, &out_planes, &pad_planes)); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 234 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
240 pad_planes = std::max<int64>( in launch()
310 const bool planes_odd = (pad_planes % 2 != 0); in launch()
394 CHECK(pad_rows >= 0 && pad_cols >= 0 && pad_planes >= 0) in launch()
396 << pad_planes << ")"; in launch()
427 .set_zero_padding(DimIndex::Z, pad_planes / 2) in launch()
490 {{pad_planes, pad_rows, pad_cols}}, in launch()
|
| D | pooling_ops_3d.h | 71 int64 pad_planes; member
|
| D | pooling_ops_3d.cc | 81 &out_plane, &pad_planes)); in Pool3dParameters()
578 const int32 pad_planes = params.pad_planes; in launch() local
607 int p_start = pp * plane_stride - pad_planes; in launch()
811 params.pad_planes, params.pad_rows, params.pad_cols, in launch()
|