Searched refs:spatial_dimension (Results 1 – 4 of 4) sorted by relevance
462 for (int64 spatial_dimension = 0; in HandleConvolution() local463 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() local488 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 …]
44 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()
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()