Searched defs:VmapInterpreterPtr (Results 1 – 1 of 1) sorted by relevance
9 struct VmapInterpreterPtr { struct10 …explicit VmapInterpreterPtr(const Interpreter* base): base_(base) { TORCH_INTERNAL_ASSERT(base->ke… in VmapInterpreterPtr() argument11 TransformType key() const { return base_->key(); } in key()12 int64_t level() const { return base_->level(); } in level()15 c10::SymInt batchSize() const { in batchSize()18 RandomnessType randomness() const { in randomness()