Searched refs:MiniArrayRef (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/ |
| H A D | arrayref_tensor.h | 15 class MiniArrayRef final {
36 /* implicit */ constexpr MiniArrayRef() : Data(nullptr), Length(0) {} in MiniArrayRef() function
40 constexpr MiniArrayRef(const T& OneElt) : Data(&OneElt), Length(1) {} in MiniArrayRef() function
43 constexpr MiniArrayRef(T* data, size_t length) : Data(data), Length(length) {} in MiniArrayRef() function
46 constexpr MiniArrayRef(T* begin, T* end) : Data(begin), Length(end - begin) {} in MiniArrayRef() function
53 /* implicit */ MiniArrayRef(Container& container) in MiniArrayRef() function
61 /* implicit */ MiniArrayRef(const std::vector<T, A>& Vec) in MiniArrayRef() function
70 /* implicit */ constexpr MiniArrayRef(std::array<T, N>& Arr) in MiniArrayRef() function
76 /* implicit */ constexpr MiniArrayRef(T (&Arr)[N]) : Data(Arr), Length(N) {} in MiniArrayRef() function
79 /* implicit */ constexpr MiniArrayRef(const volatile void* Arr) in MiniArrayRef() function
[all …]
|
| H A D | thread_local.h | 147 tensor_.set_arrayref(MiniArrayRef<T>(storage_.get(), t.numel()));
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/aoti_runtime/ |
| H A D | implementation.cpp | 54 MiniArrayRef<T>(reinterpret_cast<T*>(data_ptr), numel), in convert_handle_to_arrayref_tensor()
55 MiniArrayRef<const int64_t>(sizes, dim), in convert_handle_to_arrayref_tensor()
56 MiniArrayRef<const int64_t>(strides, dim), in convert_handle_to_arrayref_tensor()
|