Searched defs:tuple_elements (Results 1 – 9 of 9) sorted by relevance
29 std::vector<HloInstruction*> tuple_elements; in ExtractPrefix() local48 std::vector<HloInstruction*> tuple_elements; in AppendSuffix() local
210 const std::vector<HloSharding>& tuple_elements() const { in tuple_elements() function
113 std::vector<HloInstruction*> tuple_elements; in ExpandInstruction() local
89 std::vector<HloInstruction*> tuple_elements; in ConvertTupleElements() local
1476 HloInstruction::InstructionVector tuple_elements; in CloneAndFuseInternal() local
31 int64 tuple_elements = xla::ShapeUtil::TupleElementCount(shape); in PopulateInfeedLayoutVector() local135 int64 tuple_elements = xla::ShapeUtil::TupleElementCount(input_shape); in GetShapeWithLayout() local
322 std::vector<Literal> tuple_elements; in Run() local353 std::vector<Literal> tuple_elements; in Run() local
190 const int64 tuple_elements = ShapeUtil::TupleElementCount(shape); in RecursiveElementCount() local
1662 Literal tuple_elements[] = { in TEST_F() local