Searched refs:set_zero_padding (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 425 .set_zero_padding(DimIndex::X, pad_cols / 2) in launch()
426 .set_zero_padding(DimIndex::Y, pad_rows / 2) in launch()
427 .set_zero_padding(DimIndex::Z, pad_planes / 2) in launch()
|
| D | conv_grad_ops_3d.cc | 1326 .set_zero_padding(DimIndex::X, padding_cols / 2) in Compute()
1327 .set_zero_padding(DimIndex::Y, padding_rows / 2) in Compute()
1328 .set_zero_padding(DimIndex::Z, padding_planes / 2) in Compute()
1823 .set_zero_padding(DimIndex::X, padding_cols / 2) in Compute()
1824 .set_zero_padding(DimIndex::Y, padding_rows / 2) in Compute()
1825 .set_zero_padding(DimIndex::Z, padding_planes / 2) in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 405 .set_zero_padding( in GetGpuConvConfig()
433 config.conv_desc.set_zero_padding(static_cast<DimIndex>(dim), 0) in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 521 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64 value) { in set_zero_padding() function
|