Searched defs:tuple_elements (Results 1 – 13 of 13) sorted by relevance
29 std::vector<HloInstruction*> tuple_elements; in ExtractPrefix() local48 std::vector<HloInstruction*> tuple_elements; in AppendSuffix() local
300 std::vector<HloSharding>& tuple_elements() { return tuple_elements_; } in tuple_elements() function301 const std::vector<HloSharding>& tuple_elements() const { in tuple_elements() function
113 std::vector<HloInstruction*> tuple_elements; in ExpandInstruction() local
91 std::vector<HloInstruction*> tuple_elements; in ConvertTupleElements() local
1719 HloInstruction::InstructionVector tuple_elements; in CloneAndAppendInstructionIntoCalledComputation() local
31 int64_t tuple_elements = xla::ShapeUtil::TupleElementCount(shape); in PopulateInfeedLayoutVector() local177 int64_t tuple_elements = xla::ShapeUtil::TupleElementCount(input_shape); in GetShapeWithLayout() local
232 std::vector<XlaOp> tuple_elements = BuildInputs(b, str); in BuildInputs() local251 std::vector<Shape> tuple_elements = BuildOutputType(str); in BuildOutputType() local
402 std::vector<Literal> tuple_elements; in Run() local431 std::vector<Literal> tuple_elements; in Run() local
341 auto tuple_elements = out_tuple_ty->elements(); in codegen_backend_module() local
165 const auto& tuple_elements = tuple->elements(); in RemoveTupleConstants() local
134 int64_t tuple_elements = xla::ShapeUtil::TupleElementCount(*output_shape); in GetInfeedShapeWithLayout() local
208 const int64_t tuple_elements = ShapeUtil::TupleElementCount(shape); in RecursiveElementCount() local
1740 Literal tuple_elements[] = { in TEST_F() local