Searched defs:spatial_dimensions (Results 1 – 3 of 3) sorted by relevance
28 absl::Span<const int64> spatial_dimensions) in TensorFormat()
26 absl::InlinedVector<int64, 4> spatial_dimensions; in MakeNCHWFormat() local
142 absl::InlinedVector<int64, 4> spatial_dimensions(num_spatial_dims); in XlaTensorFormat() local