Searched refs:spatial_dimensions (Results 1 – 4 of 4) sorted by relevance
28 absl::Span<const int64> spatial_dimensions) in TensorFormat() argument31 spatial_dimensions_(spatial_dimensions.begin(), in TensorFormat()32 spatial_dimensions.end()) {} in TensorFormat()
26 absl::InlinedVector<int64, 4> spatial_dimensions; in MakeNCHWFormat() local28 spatial_dimensions.push_back(i + 2); in MakeNCHWFormat()31 /*spatial_dimensions=*/spatial_dimensions); in MakeNCHWFormat()
140 absl::InlinedVector<int64, 4> spatial_dimensions(num_spatial_dims); in XlaTensorFormat() local142 spatial_dimensions[spatial_dim] = in XlaTensorFormat()147 /*spatial_dimensions=*/spatial_dimensions); in XlaTensorFormat()
4810 spatial_dimensions = list(range(ndims))4822 for d in spatial_dimensions])4834 permutation = [ndims, ndims + 1] + spatial_dimensions4836 permutation = [ndims] + spatial_dimensions + [ndims + 1]