Home
last modified time | relevance | path

Searched refs:resolve_conj (Results 1 – 25 of 28) sorted by relevance

12

/external/pytorch/torch/csrc/utils/
Dtensor_list.cpp51 Tensor data = tensor.resolve_conj().resolve_neg(); in tensor_to_list()
Dtensor_numpy.cpp164 auto prepared_tensor = tensor.detach().cpu().resolve_conj().resolve_neg(); in tensor_to_numpy()
/external/pytorch/torch/onnx/_internal/
Dio_adapter.py294 torch.view_as_real(arg.resolve_conj())
500 torch.view_as_real(output.resolve_conj())
/external/pytorch/torch/onnx/_internal/exporter/
D_onnx_program.py313 torch.view_as_real(arg.resolve_conj())
/external/pytorch/aten/src/ATen/
DFunctionalInverses.cpp126 return at::view_as_real(mutated_view.resolve_conj()); in view_as_complex_inverse()
128 return at::view_as_real_copy(mutated_view.resolve_conj()); in view_as_complex_inverse()
DLegacyBatchingRegistrations.cpp1251 TRIVIAL_OP(resolve_conj); in TORCH_LIBRARY_IMPL()
/external/pytorch/aten/src/ATen/native/
DLinearAlgebra.cpp1446 c = result.resolve_conj(); in addmm_impl_cpu_()
1453 c = result.resolve_conj(); in addmm_impl_cpu_()
1457 c = result.resolve_conj().transpose(0, 1).contiguous().transpose_(0, 1); in addmm_impl_cpu_()
1471 a = m1.resolve_conj(); in addmm_impl_cpu_()
1488 b = m2.resolve_conj(); in addmm_impl_cpu_()
1878 bmm_out_or_baddbmm_(result, batch1.resolve_conj(), batch2.resolve_conj(), beta, alpha, false); in TORCH_IMPL_FUNC()
1888 …bmm_out_or_baddbmm_(result, batch1.resolve_conj(), batch2.resolve_conj(), Scalar(0.0), Scalar(1.0)… in TORCH_IMPL_FUNC()
DConvolution.cpp856 auto [i_r, i_i] = complex_to_real(input.resolve_conj()); in complex_convolution()
857 auto [w_r, w_i] = complex_to_real(weight.resolve_conj()); in complex_convolution()
873 auto [b_r, b_i] = complex_to_real(bias.resolve_conj()); in complex_convolution()
893 auto [i_r, i_i] = complex_to_real(input.resolve_conj()); in complex_convolution_mode()
894 auto [w_r, w_i] = complex_to_real(weight.resolve_conj()); in complex_convolution_mode()
903 auto [b_r, b_i] = complex_to_real(bias.resolve_conj()); in complex_convolution_mode()
DLinearAlgebraUtils.h32 return c10::MaybeOwned<Tensor>::owned(tensor.resolve_conj()); in expect_resolved_conj()
DUnaryOps.cpp646 Tensor resolve_conj(const Tensor& self) { in resolve_conj() function
DSpectralOps.cpp1055 Tensor input = at::view_as_real(self.resolve_conj()); in istft()
/external/pytorch/aten/src/ATen/native/cuda/
DBlas.cpp50 c10::MaybeOwned<Tensor> inline resolve_conj_if_indicated(const Tensor& tensor, bool resolve_conj) { in resolve_conj_if_indicated() argument
51 if (resolve_conj && tensor.is_conj()) { in resolve_conj_if_indicated()
52 return c10::MaybeOwned<Tensor>::owned(tensor.resolve_conj()); in resolve_conj_if_indicated()
/external/pytorch/functorch/op_analysis/
Dpublic_api588 resolve_conj
Dannotated_ops21 resolve_conj, complex
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesDecompositions.cpp231 OP_DECOMPOSE(resolve_conj); in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/onnx/_internal/fx/
Dfx_onnx_interpreter.py578 fake_tensor = torch.view_as_real(fake_tensor.resolve_conj())
/external/pytorch/docs/source/
Dtorch.rst572 resolve_conj
Dtensors.rst363 Tensor.resolve_conj
/external/pytorch/torch/
D_tensor_str.py346 self = self.resolve_conj()
D_torch_docs.py2962 torch.resolve_conj,
/external/pytorch/test/mobile/model_test/
Dmath_ops.py372 torch.resolve_conj(a),
/external/pytorch/test/
Dtest_numpy_interop.py184 y = x.resolve_conj()
Dtest_functionalization.py2196 return torch.view_as_real(y.resolve_conj())
/external/pytorch/torch/autograd/
Dgradcheck.py813 tv = torch.view_as_real(vJ.resolve_conj())
/external/pytorch/torch/csrc/jit/runtime/static/
Dgenerated_ops.cpp4876 aten::resolve_conj,
4883 p_node->Output(0) = at::native::resolve_conj(self); in __anon7ebb59391d002()

12