Home
last modified time | relevance | path

Searched refs:dynamic_intrusive_pointer_cast (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_distributed_ops.cpp181 c10::dynamic_intrusive_pointer_cast<dist_rpc::OwnerRRef>(rref) in __anon84498b7a0202()
184 res = c10::dynamic_intrusive_pointer_cast<dist_rpc::UserRRef>(rref) in __anon84498b7a0202()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAGraph.cpp78 dynamic_intrusive_pointer_cast<CUDAGeneratorImpl>( in register_generator_state()
H A DCUDAGeneratorImpl.cpp381 dynamic_intrusive_pointer_cast<CUDAGeneratorImpl>(gen); in graphsafe_set_state()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DQuantizedLinear.cpp431 *c10::dynamic_intrusive_pointer_cast<PackedLinearWeightFp16>( in fbgemm_linear_fp16_weight_fp32_activation()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.cpp684 c10::dynamic_intrusive_pointer_cast<torch::distributed::rpc::RRef>( in toPyObject()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Divalue_inl.h85 intrusive_ptr<T> dynamic_intrusive_pointer_cast(intrusive_ptr<U> r) { in dynamic_intrusive_pointer_cast() function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp1701 if (auto workNCCL = c10::dynamic_intrusive_pointer_cast<WorkNCCL>(work)) { in verifyWorkTimeoutForTest()