Home
last modified time | relevance | path

Searched full:resolve_conj (Results 1 – 25 of 45) sorted by relevance

12

/external/pytorch/test/profiler/
Dtest_profiler_tree.py617 aten::resolve_conj
618 aten::resolve_conj
619 aten::resolve_conj
664 aten::resolve_conj
665 aten::resolve_conj
666 aten::resolve_conj
/external/pytorch/torch/csrc/utils/
Dtensor_list.cpp51 Tensor data = tensor.resolve_conj().resolve_neg(); in tensor_to_list()
Dtensor_numpy.cpp156 "Use tensor.resolve_conj().numpy() instead."); in tensor_to_numpy()
164 auto prepared_tensor = tensor.detach().cpu().resolve_conj().resolve_neg(); in tensor_to_numpy()
/external/pytorch/aten/src/ATen/native/
DNegateFallback.cpp30 m.impl("resolve_conj", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
DComplexHelper.h60 … To resolve the conjugate tensor so you can view it as real, use self.resolve_conj(); however, be … in view_as_real()
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()
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()
DLinearAlgebraUtils.h32 return c10::MaybeOwned<Tensor>::owned(tensor.resolve_conj()); in expect_resolved_conj()
/external/pytorch/aten/src/ATen/
DConjugateFallback.cpp30 m.impl("resolve_conj", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
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()
/external/pytorch/functorch/op_analysis/
Dpublic_api588 resolve_conj
Dannotated_ops21 resolve_conj, complex
/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/docs/source/
Dtorch.rst572 resolve_conj
Dtensors.rst363 Tensor.resolve_conj
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesDecompositions.cpp231 OP_DECOMPOSE(resolve_conj); in TORCH_LIBRARY_IMPL()
/external/pytorch/test/mobile/model_test/
Dmodel_ops.yaml310 aten::resolve_conj: 12
Dcoverage.yaml516 - aten::resolve_conj
947 aten::resolve_conj: 1
Dmath_ops.py372 torch.resolve_conj(a),
/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/torch/
D_tensor_str.py346 self = self.resolve_conj()
/external/pytorch/test/
Dtest_numpy_interop.py184 y = x.resolve_conj()
/external/pytorch/torch/onnx/_internal/fx/
Dfx_onnx_interpreter.py578 fake_tensor = torch.view_as_real(fake_tensor.resolve_conj())
/external/pytorch/torch/_subclasses/
Dfunctional_tensor.py133 # return at::view_as_real(x.resolve_conj());

12