Searched refs:InferVariadicOpShape (Results 1 – 6 of 6) sorted by relevance
73 static StatusOr<Shape> InferVariadicOpShape(75 static StatusOr<Shape> InferVariadicOpShape(
1067 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference1074 return InferVariadicOpShape(opcode, operand_shapes); in InferVariadicOpShape()1077 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference2180 return InferVariadicOpShape(HloOpcode::kTuple, operand_shapes); in InferAllToAllTupleShape()
310 ShapeInference::InferVariadicOpShape(HloOpcode::kTuple, {&s32_, &f32_}); in TEST_F()2425 ShapeInference::InferVariadicOpShape(HloOpcode::kSort, {&keys, &values}); in TEST_F()2436 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()2448 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()
1805 ShapeInference::InferVariadicOpShape( in CheckVariadicShape()
1787 return ShapeInference::InferVariadicOpShape(opcode, arg_shapes); in CreateInstruction()1806 return ShapeInference::InferVariadicOpShape(opcode, arg_shapes); in CreateInstruction()
1316 ShapeInference::InferVariadicOpShape( in Tuple()2179 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferVariadicOpShape( in Sort()