Searched refs:set_dilation_rate (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 419 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
420 .set_dilation_rate(DimIndex::Y, dilations[1]) in launch()
421 .set_dilation_rate(DimIndex::Z, dilations[0]) in launch()
|
| D | conv_grad_ops_3d.cc | 1320 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1321 .set_dilation_rate(DimIndex::Y, dims.dilation(1)) in Compute()
1322 .set_dilation_rate(DimIndex::Z, dims.dilation(0)) in Compute()
1817 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1818 .set_dilation_rate(DimIndex::Y, dims.dilation(1)) in Compute()
1819 .set_dilation_rate(DimIndex::Z, dims.dilation(0)) in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 411 .set_dilation_rate( in GetGpuConvConfig()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 545 ConvolutionDescriptor& set_dilation_rate(DimIndex dim, int64 value) { in set_dilation_rate() function
|