Searched defs:spatial_dimensions (Results 1 – 6 of 6) sorted by relevance
14 const auto spatial_dimensions = static_cast<int64_t>(input_size.size()) - 2; in compute_output_size() local
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()