Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc597 for (int64 spatial_dimension = 0; in HandleConvolution() local
598 spatial_dimension < window.dimensions_size(); ++spatial_dimension) { in HandleConvolution()
601 dnums.kernel_spatial_dimensions(spatial_dimension); in HandleConvolution()
607 dnums.output_spatial_dimensions(spatial_dimension); in HandleConvolution()
613 dnums.input_spatial_dimensions(spatial_dimension); in HandleConvolution()
622 for (int64 spatial_dimension = 0; in HandleConvolution() local
623 spatial_dimension < window.dimensions_size(); ++spatial_dimension) { in HandleConvolution()
624 const auto& window_dim = window.dimensions(spatial_dimension); in HandleConvolution()
628 if (input_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
629 kernel_limits[spatial_dimension] == output_limits[spatial_dimension] && in HandleConvolution()
[all …]
Dspace_to_batch_converter.cc170 int64 spatial_dimension, int64 new_spatial_dim_size);
571 int64 spatial_dimension, int64 new_spatial_dim_size) { in IncreaseSpatialSizeOnSpaceToBatchedShape() argument
572 CHECK_EQ(batch_dimension + 1, spatial_dimension); in IncreaseSpatialSizeOnSpaceToBatchedShape()
577 int64 spatial_dim_size = activations->shape().dimensions(spatial_dimension); in IncreaseSpatialSizeOnSpaceToBatchedShape()
583 new_dimensions[spatial_dimension] = reshaped_space_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
593 padding_config.mutable_dimensions(spatial_dimension) in IncreaseSpatialSizeOnSpaceToBatchedShape()
597 padding_config.mutable_dimensions(spatial_dimension)->set_edge_padding_low(0); in IncreaseSpatialSizeOnSpaceToBatchedShape()
610 reshape_back_dims[spatial_dimension] = new_spatial_dim_size; in IncreaseSpatialSizeOnSpaceToBatchedShape()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc44 int dim = data_format.spatial_dimension(i); in AvgPoolDivideByCountWithGeneralPadding()
101 int dim = data_format.spatial_dimension(i); in MakeSpatialPaddingConfig()
184 int dim = data_format.spatial_dimension(i); in MakeSpatialPadding()
250 int dim = data_format.spatial_dimension(i); in AvgPoolGrad()
255 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()