Searched refs:set_spatial_dim (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_3d.cc | 400 .set_spatial_dim(DimIndex::X, in_cols) in launch() 401 .set_spatial_dim(DimIndex::Y, in_rows) in launch() 402 .set_spatial_dim(DimIndex::Z, in_planes) in launch() 406 .set_spatial_dim(DimIndex::X, out_cols) in launch() 407 .set_spatial_dim(DimIndex::Y, out_rows) in launch() 408 .set_spatial_dim(DimIndex::Z, out_planes) in launch() 412 filter_desc.set_spatial_dim(DimIndex::X, filter_cols) in launch() 413 .set_spatial_dim(DimIndex::Y, filter_rows) in launch() 414 .set_spatial_dim(DimIndex::Z, filter_planes) in launch()
|
D | cudnn_pooling_gpu.cc | 86 input_desc.set_spatial_dim(dim_i, in Compute() 88 output_desc.set_spatial_dim(dim_i, in Compute() 222 orig_input_desc.set_spatial_dim( in Compute() 224 orig_output_desc.set_spatial_dim(dim_i, output_size[i]); in Compute()
|
D | conv_grad_ops_3d.cc | 1300 .set_spatial_dim(DimIndex::X, compatible_input_shape.dim_size(4)) in Compute() 1301 .set_spatial_dim(DimIndex::Y, compatible_input_shape.dim_size(3)) in Compute() 1302 .set_spatial_dim(DimIndex::Z, compatible_input_shape.dim_size(2)) in Compute() 1307 .set_spatial_dim(DimIndex::X, dims.output_size(2)) in Compute() 1308 .set_spatial_dim(DimIndex::Y, dims.output_size(1)) in Compute() 1309 .set_spatial_dim(DimIndex::Z, dims.output_size(0)) in Compute() 1313 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) in Compute() 1314 .set_spatial_dim(DimIndex::Y, dims.filter_size(1)) in Compute() 1315 .set_spatial_dim(DimIndex::Z, dims.filter_size(0)) in Compute() 1794 .set_spatial_dim(DimIndex::X, in Compute() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_runner.cc | 382 input_descriptor.set_spatial_dim( in GetGpuConvConfig() 395 filter_descriptor.set_spatial_dim( in GetGpuConvConfig() 423 output_descriptor.set_spatial_dim( in GetGpuConvConfig() 430 input_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig() 431 output_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig() 432 filter_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 287 BatchDescriptor& set_spatial_dim(DimIndex dim, int64 value) { in set_spatial_dim() function 411 FilterDescriptor& set_spatial_dim(DimIndex dim, int64 value) { in set_spatial_dim() function
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 4641 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 4608 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()
|