Home
last modified time | relevance | path

Searched refs:VmapInterpreterMeta (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/functorch/
DInterpreter.h91 struct VmapInterpreterMeta { struct
92 explicit VmapInterpreterMeta(c10::SymInt batchSize, RandomnessType randomness) : in VmapInterpreterMeta() function
118 VmapInterpreterMeta, argument
126 …return Interpreter(TransformType::Vmap, level, VmapInterpreterMeta(std::move(batchSize), randomnes… in Vmap()
DVmapInterpreter.h16 return std::get<VmapInterpreterMeta>(base_->meta()).batchSize_; in batchSize()
19 return std::get<VmapInterpreterMeta>(base_->meta()).randomness_; in randomness()
DInterpreter.cpp107 TORCH_INTERNAL_ASSERT(std::holds_alternative<VmapInterpreterMeta>(this->meta()));\