Searched defs:spatial_dimensions (Results 1 – 5 of 5) sorted by relevance
28 absl::Span<const int64_t> spatial_dimensions) in TensorFormat()
26 absl::InlinedVector<int64_t, 4> spatial_dimensions; in MakeNCHWFormat() local
150 absl::InlinedVector<int64_t, 4> spatial_dimensions(num_spatial_dims); in XlaTensorFormat() local
745 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions) { in SplitAndTransposeMergedBatch()797 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions, in ChangeSpatialSizeOnSpaceToBatchedShape()
540 ArrayRef<int64_t> spatial_dimensions, in FormatToNHWC()