Searched defs:tuple_elements (Results 1 – 10 of 10) sorted by relevance
29 std::vector<HloInstruction*> tuple_elements; in ExtractPrefix() local48 std::vector<HloInstruction*> tuple_elements; in AppendSuffix() local
267 std::vector<HloSharding>& tuple_elements() { return tuple_elements_; } in tuple_elements() function268 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
1745 HloInstruction::InstructionVector tuple_elements; in CloneAndFuseInternal() local
31 int64 tuple_elements = xla::ShapeUtil::TupleElementCount(shape); in PopulateInfeedLayoutVector() local178 int64 tuple_elements = xla::ShapeUtil::TupleElementCount(input_shape); in GetShapeWithLayout() local
364 std::vector<Literal> tuple_elements; in Run() local394 std::vector<Literal> tuple_elements; in Run() local
134 int64 tuple_elements = xla::ShapeUtil::TupleElementCount(*output_shape); in GetInfeedShapeWithLayout() local
205 const int64 tuple_elements = ShapeUtil::TupleElementCount(shape); in RecursiveElementCount() local
1731 Literal tuple_elements[] = { in TEST_F() local