Searched defs:self_tensor (Results 1 – 7 of 7) sorted by relevance
80 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90202() local92 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90302() local110 exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>(); in __anon9c72e3c90502() local
120 void _assert_async_msg_cuda(const Tensor& self_tensor, c10::string_view assert_msg) { in _assert_async_msg_cuda()137 void _assert_async_cuda(const Tensor& self_tensor) { in _assert_async_cuda()
68 auto self_tensor = torch::lazy::TryGetLtcTensor(self); in _copy_from() local119 auto self_tensor = torch::lazy::TryGetLtcTensor(self); in _copy_from_and_resize() local321 auto self_tensor = torch::lazy::TryGetLtcTensor(self); in fill_() local
47 Tensor self_tensor = tf.ones({3, 5}); in TEST_F() local69 Tensor self_tensor = tf.ones({3, 5}); in TEST_F() local268 Tensor self_tensor = tf.ones({1}); in TEST_F() local
18 def _tensor_version_fake(fake_mode, self_tensor): argument
1037 auto self_tensor = node->inputs()[0]; // input tensor in typesAreSupported() local
4940 def __init__(self, self_tensor, storage_tensor): argument