Home
last modified time | relevance | path

Searched defs:OperatorMap (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dir.h1655 struct OperatorMap { struct
1661 OperatorMap() = default; argument
1663 explicit OperatorMap( in OperatorMap() argument
1668 explicit OperatorMap(std::initializer_list<std::pair<const char*, T>> init) { in OperatorMap() argument
1672 void insert(const std::shared_ptr<Operator>& op, T val) { in insert()
1679 void insert(const OperatorSet& op_set, T val) { in insert()
1685 void insert( in insert()
1692 void insert(std::initializer_list<std::pair<const char*, T>> v) { in insert()
1698 void erase(const std::shared_ptr<Operator>& op) { in erase()
1714 bool contains(const Operator& op) const { in contains()
[all …]
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_ir.cpp159 TEST(IRTest, OperatorMap) { in TEST() argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Doperator.cpp15 using OperatorMap = typedef