Searched defs:operand_shapes (Results 1 – 7 of 7) sorted by relevance
12 repeated xla.ShapeProto operand_shapes = 2; field
121 std::vector<const Shape*> operand_shapes; in HandleConcatenate() local188 std::vector<const Shape*> operand_shapes; in HandleAllReduce() local196 std::vector<const Shape*> operand_shapes; in HandleAllToAll() local448 std::vector<const Shape*> operand_shapes; in HandleReduce() local600 std::vector<const Shape*> operand_shapes; in HandleMap() local828 std::vector<const Shape*> operand_shapes; in HandleAfterAll() local
207 std::vector<const Shape*> operand_shapes; in MakeConcatHlo() local233 std::vector<const Shape*> operand_shapes; in MakeMapHlo() local
1023 std::vector<const Shape*> operand_shapes; in InferVariadicOpShape() local1032 HloOpcode opcode, absl::Span<const Shape* const> operand_shapes) { in InferVariadicOpShape()1976 absl::Span<const Shape* const> operand_shapes) { in InferAllReduceShape()2019 absl::Span<const Shape* const> operand_shapes) { in InferAllToAllTupleShape()
1691 absl::InlinedVector<const Shape*, 1> operand_shapes; in HandleReduce() local
448 std::vector<Shape> operand_shapes; in CreateFromProto() local
113 std::vector<Shape> operand_shapes; in GetOperandShapes() local