Home
last modified time | relevance | path

Searched defs:tensor_impl (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/executorch/extension/tensor/test/
H A Dtensor_impl_ptr_test.cpp28 auto tensor_impl = make_tensor_impl_ptr({}, &scalar_data); in TEST_F() local
39 auto tensor_impl = make_tensor_impl_ptr({}, {3.14f}); in TEST_F() local
50 auto tensor_impl = make_tensor_impl_ptr({4, 5}, data, {0, 1}, {5, 1}); in TEST_F() local
77 auto tensor_impl = make_tensor_impl_ptr({3, 4}, data, {}, {4, 1}); in TEST_F() local
89 auto tensor_impl = make_tensor_impl_ptr({3, 4}, data, {}, {1, 3}); in TEST_F() local
100 auto tensor_impl = make_tensor_impl_ptr({2, 3, 4}, data); in TEST_F() local
119 auto tensor_impl = make_tensor_impl_ptr({3, 4}, data, {1, 0}, {1, 3}); in TEST_F() local
139 auto tensor_impl = make_tensor_impl_ptr({4, 5}, data); in TEST_F() local
152 auto tensor_impl = make_tensor_impl_ptr( in TEST_F() local
169 auto tensor_impl = make_tensor_impl_ptr( in TEST_F() local
[all …]
/aosp_15_r20/external/executorch/extension/aten_util/test/
H A Daten_bridge_test.cpp42 torch::executor::TensorImpl tensor_impl( in TEST() local
62 std::unique_ptr<torch::executor::TensorImpl> tensor_impl = in TEST() local
94 torch::executor::TensorImpl tensor_impl( in TEST() local
119 torch::executor::TensorImpl tensor_impl( in TEST() local
139 torch::executor::TensorImpl tensor_impl( in TEST() local
/aosp_15_r20/external/executorch/extension/tensor/
H A Dtensor_ptr.cpp28 exec_aten::TensorImpl tensor_impl; member
93 exec_aten::TensorImpl tensor_impl( in make_tensor_ptr() local
121 auto tensor_impl = c10::make_intrusive<exec_aten::TensorImpl>( in make_tensor_ptr() local
H A Dtensor_impl_ptr.cpp92 auto tensor_impl = std::make_unique<exec_aten::TensorImpl>( in make_tensor_impl_ptr() local
122 auto tensor_impl = c10::make_intrusive<at::TensorImpl>( in make_tensor_impl_ptr() local
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DTensorBody.h107 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) in Tensor()
119 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) { in wrap_tensor_impl()
H A DFunctions.cpp46 TensorImpl* tensor_impl = tensor.unsafeGetTensorImpl(); in make_tensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensorBase.h99 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) in TensorBase()
112 c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl> tensor_impl) { in wrap_tensor_impl()
H A DUnsafeFromTH.h7 …auto tensor_impl = c10::intrusive_ptr<TensorImpl, UndefinedTensorImpl>::reclaim(static_cast<Tensor… in unsafeTensorFromTH() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dlazy_tensor_test.cpp6 auto tensor_impl = in LazyTensorTest() local
H A Dxla_tensor_test.cpp34 auto tensor_impl = c10::make_intrusive<TensorImpl, UndefinedTensorImpl>( in TEST() local
H A Dextension_backend_test.cpp21 auto tensor_impl = c10::make_intrusive<TensorImpl, UndefinedTensorImpl>( in empty_override() local
/aosp_15_r20/external/executorch/runtime/executor/
H A Dtensor_parser_portable.cpp116 auto* tensor_impl = in parseTensor() local
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dopen_registration_extension.cpp340 c10::TensorImpl* tensor_impl = tensor.unsafeGetTensorImpl(); in unsafe_create_cpu_tensor_from_dummy_tensor() local
423 at::TensorImpl* tensor_impl = self.unsafeGetTensorImpl(); in custom_resize_() local
H A Dmaia_extension.cpp9 auto tensor_impl = c10::make_intrusive<TensorImpl, UndefinedTensorImpl>( in get_tensor() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dbroadcast_util.cpp58 auto tensor_impl = static_cast<TensorImpl*>(malloc(sizeof(TensorImpl))); in make_tensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerUtils.cpp177 const auto* tensor_impl = get_nested_tensor_impl(input_nestedtensor); in view_as_dense() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_variable.cpp374 c10::TensorImpl* tensor_impl = tensor.unsafeGetTensorImpl(); in THPVariable_tryResurrect() local
855 TensorImpl* tensor_impl = tensor.unsafeGetTensorImpl(); in THPVariable_make_wrapper_subclass() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp363 const auto tensor_impl = tensor.unsafeGetTensorImpl(); in convertIValue() local