Searched refs:_efficientzerotensor (Results 1 – 16 of 16) sorted by relevance
| /external/pytorch/torch/_lazy/ |
| D | tensor_factory_functions.py | 36 torch._efficientzerotensor,
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Blas.cpp | 176 return at::_efficientzerotensor({}, self.options()); in dot()
214 return at::_efficientzerotensor({}, self.options()); in vdot()
|
| D | BinaryOps.cpp | 1018 return at::_efficientzerotensor(meta_out.sizes(), meta_out.options().device(out_device)); in mul_zerotensor()
1035 return at::_efficientzerotensor(meta_out.sizes(), meta_out.options().device(out_device)); in div_zerotensor()
1045 return at::_efficientzerotensor(meta_out.sizes(), meta_out.options().device(out_device)); in div_zerotensor()
1066 return at::_efficientzerotensor(meta_out.sizes(), meta_out.options().device(out_device)); in maybe_add_maybe_sub()
1097 return at::_efficientzerotensor( in linalg_cross_zerotensor()
|
| D | TensorFactories.cpp | 1321 Tensor _efficientzerotensor(IntArrayRef size, in _efficientzerotensor() function
|
| /external/pytorch/torch/_export/passes/ |
| D | constant_folding.py | 136 and node.target == aten._efficientzerotensor.default
|
| /external/pytorch/torch/_inductor/ |
| D | constant_folding.py | 159 and node.target == aten._efficientzerotensor.default
|
| D | lowering.py | 2360 make_fallback(aten._efficientzerotensor)
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Blas.cpp | 684 return at::_efficientzerotensor({}, self.options()); in dot_cuda()
727 return at::_efficientzerotensor({}, self.options()); in vdot_cuda()
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 95 aten::_efficientzerotensor
96 aten::_efficientzerotensor.out
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 584 return at::_efficientzerotensor(sgn.sizes(), sgn.options()); in sgn_backward()
2372 return at::_efficientzerotensor(input.sizes(), input.options()); in binary_cross_entropy_with_logits_backward()
2426 return at::_efficientzerotensor(target.sizes(), target.options()); in binary_cross_entropy_with_logits_target_backward()
4286 at::_efficientzerotensor(sign.sizes(), sign.options()), trAinvE); in slogdet_jvp()
6006 : at::_efficientzerotensor(t.sizes(), t.options())); in cat_jvp()
6031 : at::_efficientzerotensor(t.sizes(), t.options())); in block_diag_jvp()
6057 : at::_efficientzerotensor(t.sizes(), t.options())); in stack_jvp()
|
| /external/pytorch/test/ |
| D | test_python_dispatch.py | 234 y = torch._efficientzerotensor(2)
476 return torch._efficientzerotensor(args[0].shape)
480 y = torch._efficientzerotensor(3)
|
| D | test_fake_tensor.py | 1864 x = torch._efficientzerotensor(3)
1868 y = torch._efficientzerotensor(3)
|
| D | test_functionalization.py | 2143 y = torch.ops.aten._efficientzerotensor([4])
|
| D | test_serialization.py | 1157 t = torch._efficientzerotensor((4, 5))
|
| D | test_torch.py | 6634 y = torch._efficientzerotensor(3, device=device)
|
| /external/pytorch/test/cpp/api/ |
| D | serialize.cpp | 302 auto t = torch::_efficientzerotensor({5, 5}); in TEST()
|