Searched refs:spatial_dimension (Results 1 – 5 of 5) sorted by relevance
597 for (int64 spatial_dimension = 0; in HandleConvolution() local598 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() local623 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 …]
170 int64 spatial_dimension, int64 new_spatial_dim_size);571 int64 spatial_dimension, int64 new_spatial_dim_size) { in IncreaseSpatialSizeOnSpaceToBatchedShape() argument572 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()
44 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()
38 int spatial_dimension(int dim) const { return spatial_dimensions_[dim]; } in spatial_dimension() function
54 int dim = data_format.spatial_dimension(i); in ExpandWithBatchAndFeatureDimensions()