Home
last modified time | relevance | path

Searched refs:spatial_dimensions (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.cpp14 const auto spatial_dimensions = static_cast<int64_t>(input_size.size()) - 2; in compute_output_size() local
17 TORCH_CHECK(static_cast<int64_t>(output_size->size()) == spatial_dimensions); in compute_output_size()
22 TORCH_CHECK(static_cast<int64_t>(scale_factors->size()) == spatial_dimensions); in compute_output_size()
24 for (const auto i : c10::irange(spatial_dimensions)) { in compute_output_size()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dpooling.h28 absl::Span<const int64_t> spatial_dimensions) in TensorFormat() argument
31 spatial_dimensions_(spatial_dimensions.begin(), in TensorFormat()
32 spatial_dimensions.end()) {} in TensorFormat()
H A Dpooling_test.cc26 absl::InlinedVector<int64_t, 4> spatial_dimensions; in MakeNCHWFormat() local
28 spatial_dimensions.push_back(i + 2); in MakeNCHWFormat()
31 /*spatial_dimensions=*/spatial_dimensions); in MakeNCHWFormat()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc220 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions);
745 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions) { in SplitAndTransposeMergedBatch() argument
746 CHECK_EQ(batch_dimension + 1, spatial_dimensions[0]); in SplitAndTransposeMergedBatch()
758 const int64_t spatial_dim_count = spatial_dimensions.size(); in SplitAndTransposeMergedBatch()
761 new_dimensions.insert(new_dimensions.begin() + spatial_dimensions[0], in SplitAndTransposeMergedBatch()
797 int64_t old_batch_size, absl::Span<const int64_t> spatial_dimensions, in ChangeSpatialSizeOnSpaceToBatchedShape() argument
799 CHECK_EQ(batch_dimension + 1, spatial_dimensions[0]); in ChangeSpatialSizeOnSpaceToBatchedShape()
803 const int64_t spatial_dim_count = spatial_dimensions.size(); in ChangeSpatialSizeOnSpaceToBatchedShape()
805 activations->shape().dimensions(spatial_dimensions[0]); in ChangeSpatialSizeOnSpaceToBatchedShape()
812 spatial_dimensions)); in ChangeSpatialSizeOnSpaceToBatchedShape()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc150 absl::InlinedVector<int64_t, 4> spatial_dimensions(num_spatial_dims); in XlaTensorFormat() local
152 spatial_dimensions[spatial_dim] = in XlaTensorFormat()
157 /*spatial_dimensions=*/spatial_dimensions); in XlaTensorFormat()
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py5866 spatial_dimensions = list(range(ndims))
5878 kernel_size[d]) for d in spatial_dimensions)
5890 permutation = [ndims, ndims + 1] + spatial_dimensions
5892 permutation = [ndims] + spatial_dimensions + [ndims + 1]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlegalize_hlo.cc540 ArrayRef<int64_t> spatial_dimensions, in FormatToNHWC() argument
546 const int spatial_dim_start = spatial_dimensions.front(); in FormatToNHWC()
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py2731 spatial_dimensions = len(input_size) - 2
2737 torch._check(len(output_size) == spatial_dimensions, lambda: "")
2745 torch._check(len(scale_factors) == spatial_dimensions, lambda: "")