Home
last modified time | relevance | path

Searched refs:TensorImplAddress (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/profiler/
Ddata_flow.h41 using TensorImplAddress = strong::type< variable
78 return TensorImplAddress{weak_self_._unsafe_get_target()}; in get()
Ddata_flow.cpp9 static constexpr TensorImplAddress NoTensorImpl{nullptr};
13 TensorImplAddress impl_;
150 ska::flat_hash_map<TensorImplAddress, AllocationID> impl_map; in calculateUniqueTensorIDs()
Dcollection.h76 TensorImplAddress impl() const { in impl()
/external/pytorch/torch/csrc/profiler/python/
Dinit.h17 STRONG_POINTER_TYPE_CASTER(torch::profiler::impl::TensorImplAddress);