Searched refs:InferVariadicOpShape (Results 1 – 6 of 6) sorted by relevance
75 static StatusOr<Shape> InferVariadicOpShape(77 static StatusOr<Shape> InferVariadicOpShape(
1065 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference1072 return InferVariadicOpShape(opcode, operand_shapes); in InferVariadicOpShape()1075 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference2104 return InferVariadicOpShape(HloOpcode::kTuple, operand_shapes); in InferAllToAllTupleShape()
308 ShapeInference::InferVariadicOpShape(HloOpcode::kTuple, {&s32_, &f32_}); in TEST_F()2370 ShapeInference::InferVariadicOpShape(HloOpcode::kSort, {&keys, &values}); in TEST_F()2381 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()2393 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()
1271 ShapeInference::InferVariadicOpShape( in CheckVariadicShape()
1413 return ShapeInference::InferVariadicOpShape(opcode, arg_shapes); in ParseInstructionRhs()
1243 ShapeInference::InferVariadicOpShape( in Tuple()2068 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferVariadicOpShape( in Sort()