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 __anon4901d40f0302()
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 __anon6fb97db50302()
657 ….SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { return {2}; }); in __anon41f1f9b90702()668 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { in __anon41f1f9b90902()1390 .SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { in __anon41f1f9b90b02()
302 ….SetInfer([](const ShapeArray &, const HashSet<size_t> &) -> std::vector<int64_t> { return {1}; }); in __anonbbaa5cbc0702()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