Searched refs:VmapInterpreterMeta (Results 1 – 3 of 3) sorted by relevance
91 struct VmapInterpreterMeta { struct92 explicit VmapInterpreterMeta(c10::SymInt batchSize, RandomnessType randomness) : in VmapInterpreterMeta() function118 VmapInterpreterMeta, argument126 …return Interpreter(TransformType::Vmap, level, VmapInterpreterMeta(std::move(batchSize), randomnes… in Vmap()
16 return std::get<VmapInterpreterMeta>(base_->meta()).batchSize_; in batchSize()19 return std::get<VmapInterpreterMeta>(base_->meta()).randomness_; in randomness()
107 TORCH_INTERNAL_ASSERT(std::holds_alternative<VmapInterpreterMeta>(this->meta()));\