| /external/pytorch/aten/src/ATen/core/ |
| 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 …]
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | shape.h | 63 std::vector<int64_t> sizes_; variable
|
| /external/executorch/runtime/executor/ |
| D | method_meta.h | 81 Span<const int32_t> sizes_; variable
|
| /external/pytorch/torch/csrc/jit/mobile/nnc/ |
| D | context.h | 30 std::vector<int64_t> sizes_; member
48 std::vector<int64_t> sizes_; member
|
| /external/pytorch/torch/csrc/dynamo/ |
| D | guards.h | 83 std::vector<std::optional<c10::SymInt>> sizes_; variable
|
| /external/pytorch/aten/src/ATen/ops/ |
| D | from_blob.h | 92 IntArrayRef sizes_; variable
|
| /external/pytorch/aten/src/ATen/ |
| D | CPUApplyUtils.h | 63 int64_t sizes_[N] = {0}; member
92 std::vector<int64_t> sizes_; member
|
| D | FunctionalTensorWrapper.cpp | 226 auto sizes_ = value_.sym_sizes(); in replace_() local
278 auto sizes_ = value_.sym_sizes(); in set__impl() local
|
| /external/pytorch/torch/csrc/inductor/aoti_eager/ |
| D | kernel_meta_info.h | 44 std::vector<int64_t> sizes_; member
|
| D | kernel_meta_info.cpp | 67 auto sizes_ = c10::IntArrayRef(other.sizes_); in operator ==() local
|
| /external/pytorch/aten/src/ATen/cuda/detail/ |
| D | OffsetCalculator.cuh | 72 at::cuda::detail::IntDivider<index_t> sizes_[MAX_DIMS]; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization.h | 175 RematerializationSizes* sizes_; variable
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | SpectralOps.cu | 23 at::cuda::detail::IntDivider<index_t> sizes_[MAX_DIMS]; member
|
| D | CuFFTPlanCache.h | 35 int64_t sizes_[max_rank + 1]; member
|
| /external/pytorch/torch/csrc/profiler/ |
| D | collection.h | 82 std::vector<int64_t> sizes_; member
|
| /external/executorch/backends/vulkan/runtime/api/containers/ |
| D | Tensor.h | 241 std::vector<int64_t> sizes_; variable
|
| /external/pytorch/torch/csrc/api/include/torch/detail/ |
| D | TensorDataContainer.h | 346 std::vector<int64_t> sizes_; member
|
| /external/pytorch/aten/src/ATen/native/vulkan/api/ |
| D | Tensor.h | 144 std::vector<int64_t> sizes_; variable
|
| /external/pytorch/torch/csrc/jit/codegen/onednn/ |
| D | LlgaTensorImpl.h | 231 std::vector<int64_t> sizes_; member
|
| /external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| D | arrayref_tensor.h | 263 MiniArrayRef<const int64_t> sizes_; variable
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | graph_executor.cpp | 196 std::vector<size_t> sizes_; member
242 std::vector<size_t> sizes_; member
|
| /external/libvpx/test/ |
| D | vp9_ratectrl_rtc_test.cc | 640 uint32_t sizes_[8]; member in __anon1dfb5f140111::RcInterfaceSvcTest
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api.cc | 162 const ObjectsSizes sizes_; member in tflite::gpu::gl::__anonbd8198700111::InferenceContextWithBatchImpl
|
| /external/pytorch/torch/csrc/distributed/c10d/ |
| D | NCCLUtils.hpp | 633 c10::SmallVector<int64_t, 8> sizes_; // flattened from inputs, outputs member
|
| /external/pytorch/test/cpp_extensions/ |
| D | open_registration_extension.cpp | 320 const auto& sizes_ = src.sizes(); in unsafe_create_cpu_tensor_from_dummy_tensor() local
|