| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 360 at::Tensor* self_tensor = tensor_handle_to_tensor_pointer(self); in aoti_torch__reinterpret_tensor()
364 *self_tensor, sizes, strides, offset_increment)); in aoti_torch__reinterpret_tensor()
695 at::Tensor* self_tensor = tensor_handle_to_tensor_pointer(self); in aoti_torch__scaled_mm()
703 *self_tensor, in aoti_torch__scaled_mm()
726 at::Tensor* self_tensor = tensor_handle_to_tensor_pointer(self); in aoti_torch__scaled_mm_v2()
734 *self_tensor, in aoti_torch__scaled_mm_v2()
775 at::Tensor* self_tensor = tensor_handle_to_tensor_pointer(self); in aoti_torch_clone()
776 *ret = new_tensor_handle(self_tensor->clone()); in aoti_torch_clone()
790 at::Tensor* self_tensor = tensor_handle_to_tensor_pointer(self); in aoti_torch_addmm_out()
794 *out_tensor, *self_tensor, *mat1_tensor, *mat2_tensor, beta, alpha); in aoti_torch_addmm_out()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 68 auto self_tensor = torch::lazy::TryGetLtcTensor(self); in _copy_from() local
69 if (!self_tensor) { in _copy_from()
79 TORCH_CHECK(self_tensor); in _copy_from()
80 at::Tensor tensor = self_tensor->ToTensor(/*detached=*/false); in _copy_from()
91 auto src_tensor_data = self_tensor->CurrentTensorData(); in _copy_from()
102 dst_tensor_data->copy_(self_tensor->ToTensor(/*detached=*/false)); in _copy_from()
105 copy_(dst_tensor, self_tensor); in _copy_from()
119 auto self_tensor = torch::lazy::TryGetLtcTensor(self); in _copy_from_and_resize() local
120 if (!self_tensor) { in _copy_from_and_resize()
124 TORCH_CHECK(self_tensor); in _copy_from_and_resize()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/prim_ops/ |
| H A D | register_prim_ops.cpp | 80 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90202() local
82 int64_t size = self_tensor.size(dim_val); in __anon9c72e3c90202()
92 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90302() local
95 self_tensor.scalar_type(), in __anon9c72e3c90302()
100 out = EValue(Scalar(self_tensor.const_data_ptr<CTYPE>()[0])); in __anon9c72e3c90302()
110 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90502() local
111 int64_t numel = self_tensor.numel(); in __anon9c72e3c90502()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorCompare.cu | 120 void _assert_async_msg_cuda(const Tensor& self_tensor, c10::string_view assert_msg) { in _assert_async_msg_cuda() argument
121 const TensorBase &self = get_tensor_base(self_tensor); in _assert_async_msg_cuda()
137 void _assert_async_cuda(const Tensor& self_tensor) { in _assert_async_cuda() argument
138 _assert_async_msg_cuda(self_tensor, ""); in _assert_async_cuda()
|
| /aosp_15_r20/external/executorch/kernels/prim_ops/test/ |
| H A D | prim_ops_test.cpp | 47 Tensor self_tensor = tf.ones({3, 5}); in TEST_F() local
51 values[0] = EValue(self_tensor); in TEST_F()
69 Tensor self_tensor = tf.ones({3, 5}); in TEST_F() local
72 values[0] = EValue(self_tensor); in TEST_F()
268 Tensor self_tensor = tf.ones({1}); in TEST_F() local
272 values[0] = EValue(self_tensor); in TEST_F()
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_multi_threaded_pg.py | 250 self_tensor = torch.rand(3, 3)
251 rank_0_tensor = self_tensor.clone()
253 self.assertEqualOnRank(rank_0_tensor, self_tensor, rank=0)
254 self.assertNotEqualOnRank(rank_0_tensor, self_tensor, rank=1)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 229 …m.impl("repeat_interleave.self_Tensor", static_cast<decltype(&ATEN_FN2(repeat_interleave, self_Ten… in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_common_rules.py | 296 self_tensor = DTensorSpec.from_dim_map(
304 OpSchema(where_call, (condition, self_tensor, other_tensor), {})
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | tensor_version_op.py | 18 def _tensor_version_fake(fake_mode, self_tensor): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NegateFallback.cpp | 32 m.impl("repeat_interleave.self_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| H A D | native_functions.yaml | 4865 - func: repeat_interleave.self_Tensor(Tensor self, Tensor repeats, int? dim=None, *, SymInt? output…
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ConjugateFallback.cpp | 33 m.impl("repeat_interleave.self_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 64 aten.repeat_interleave.self_Tensor,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | jit_metaprogramming_utils.py | 519 self_tensor = deepcopy(self_variable.data)
523 f_args_tensor = (self_tensor,) + args_tensor
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 302 aten::repeat_interleave.self_Tensor: 1
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | tensorexpr_fuser.cpp | 1037 auto self_tensor = node->inputs()[0]; // input tensor in typesAreSupported() local
1039 if (auto const& tt = self_tensor->type()->cast<TensorType>()) { in typesAreSupported()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 4940 def __init__(self, self_tensor, storage_tensor): argument
4941 self_tensor.freeze_layout()
4943 self_tensor.get_layout(),
4944 [self_tensor, storage_tensor],
4947 V.graph.never_reuse_buffers.add(self_tensor.data.get_name())
4950 mark_node_as_mutating(self, self_tensor, storage_tensor)
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 2337 self_tensor: torch.Tensor,
2342 self_tensor, indices, values, accumulate=True
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_misc.py | 4281 # breaks on self_Tensor
|