| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | TensorAccessor.h | 41 const index_t* sizes_, in TensorAccessorBase()
64 const index_t* sizes_; variable
79 const index_t* sizes_, in TensorAccessor()
99 const index_t* sizes_, in TensorAccessor()
127 const index_t* sizes_, in GenericPackedTensorAccessorBase()
139 const source_index_t* sizes_, in GenericPackedTensorAccessorBase()
163 index_t sizes_[N]; variable
183 const index_t* sizes_, in GenericPackedTensorAccessor()
191 const source_index_t* sizes_, in GenericPackedTensorAccessor()
230 const index_t* sizes_, in GenericPackedTensorAccessor()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/nnc/ |
| H A D | context.h | 30 std::vector<int64_t> sizes_; member
48 std::vector<int64_t> sizes_; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | CPUApplyUtils.h | 63 int64_t sizes_[N] = {0}; member
92 std::vector<int64_t> sizes_; member
|
| H A D | FunctionalTensorWrapper.cpp | 226 auto sizes_ = value_.sym_sizes(); in replace_() local
278 auto sizes_ = value_.sym_sizes(); in set__impl() local
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | method_meta.h | 81 Span<const int32_t> sizes_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | shape.h | 63 std::vector<int64_t> sizes_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
| H A D | guards.h | 83 std::vector<std::optional<c10::SymInt>> sizes_; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ops/ |
| H A D | from_blob.h | 92 IntArrayRef sizes_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_eager/ |
| H A D | kernel_meta_info.h | 44 std::vector<int64_t> sizes_; member
|
| H A D | kernel_meta_info.cpp | 67 auto sizes_ = c10::IntArrayRef(other.sizes_); in operator ==() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/ |
| H A D | OffsetCalculator.cuh | 72 at::cuda::detail::IntDivider<index_t> sizes_[MAX_DIMS]; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SpectralOps.cu | 23 at::cuda::detail::IntDivider<index_t> sizes_[MAX_DIMS]; member
|
| H A D | CuFFTPlanCache.h | 35 int64_t sizes_[max_rank + 1]; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_rematerialization.h | 175 RematerializationSizes* sizes_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | graph_executor.cpp | 196 std::vector<size_t> sizes_; member
242 std::vector<size_t> sizes_; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/ |
| H A D | collection.h | 82 std::vector<int64_t> sizes_; member
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/containers/ |
| H A D | Tensor.h | 241 std::vector<int64_t> sizes_; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/ |
| H A D | Tensor.h | 144 std::vector<int64_t> sizes_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/detail/ |
| H A D | TensorDataContainer.h | 346 std::vector<int64_t> sizes_; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | LlgaTensorImpl.h | 231 std::vector<int64_t> sizes_; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| H A D | arrayref_tensor.h | 263 MiniArrayRef<const int64_t> sizes_; variable
|
| /aosp_15_r20/external/libvpx/test/ |
| H A D | vp9_ratectrl_rtc_test.cc | 640 uint32_t sizes_[8]; member in __anonae8f34940111::RcInterfaceSvcTest
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| H A D | api.cc | 162 const ObjectsSizes sizes_; member in tflite::gpu::gl::__anon655d0df00111::InferenceContextWithBatchImpl
|
| /aosp_15_r20/art/test/906-iterate-heap/ |
| H A D | iterate_heap.cc | 133 std::vector<jlong> sizes_; in Java_art_Test906_iterateThroughHeapData() member in art::Test906IterateHeap::Java_art_Test906_iterateThroughHeapData::DataIterationConfig
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | NCCLUtils.hpp | 633 c10::SmallVector<int64_t, 8> sizes_; // flattened from inputs, outputs member
|