Home
last modified time | relevance | path

Searched refs:spatial_dimension (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc462 for (int64 spatial_dimension = 0; in HandleConvolution() local
463 spatial_dimension < window.dimensions_size(); ++spatial_dimension) { in HandleConvolution()
466 dnums.kernel_spatial_dimensions(spatial_dimension); in HandleConvolution()
472 dnums.output_spatial_dimensions(spatial_dimension); in HandleConvolution()
478 dnums.input_spatial_dimensions(spatial_dimension); in HandleConvolution()
487 for (int64 spatial_dimension = 0; in HandleConvolution() local
488 spatial_dimension < window.dimensions_size(); ++spatial_dimension) { in HandleConvolution()
491 for (int64 kernel_idx = 0; kernel_idx < kernel_limits[spatial_dimension]; in HandleConvolution()
494 for (int64 output_idx = 0; output_idx < output_limits[spatial_dimension]; in HandleConvolution()
498 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc44 int dim = data_format.spatial_dimension(i); in AvgPoolDivideByCountWithGeneralPadding()
100 int dim = data_format.spatial_dimension(i); in MakeSpatialPaddingConfig()
183 int dim = data_format.spatial_dimension(i); in MakeSpatialPadding()
247 int dim = data_format.spatial_dimension(i); in AvgPoolGrad()
252 int dim = data_format.spatial_dimension(i); in AvgPoolGrad()
Dpooling.h38 int spatial_dimension(int dim) const { return spatial_dimensions_[dim]; } in spatial_dimension() function
Dpooling_test.cc54 int dim = data_format.spatial_dimension(i); in ExpandWithBatchAndFeatureDimensions()