Searched defs:ShapeArray (Results 1 – 5 of 5) sorted by relevance
25 using ShapeArray = std::vector<ShapeVector>; variable
121 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> ShapeVector { return {1}; }); in __anonadbcbaed0302()
64 std::vector<int64_t> Infer(const ShapeArray &, const HashSet<size_t> &) const override { in Infer() argument180 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> ShapeVector { return {1}; }); in __anon7f04a7930302()
657 ….SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { return {2}; }); in __anon4177fe570702()668 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { in __anon4177fe570902()1390 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { in __anon4177fe570b02()
302 ….SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { return {1}; }); in __anond7f8849a0702()422 …std::vector<int64_t> Infer(const ShapeArray &, const HashSet<size_t> &) const override { return {1… in Infer() argument447 …std::vector<int64_t> Infer(const ShapeArray &, const HashSet<size_t> &) const override { return {2… in Infer() argument