Searched refs:InferUnaryOpShape (Results 1 – 5 of 5) sorted by relevance
49 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,51 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
225 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape() function in xla::ShapeInference227 return InferUnaryOpShape(opcode, operand->shape()); in InferUnaryOpShape()230 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape() function in xla::ShapeInference
918 ShapeInference::InferUnaryOpShape(instruction->opcode(), in CheckUnaryShape()
106 ShapeInference::InferUnaryOpShape(HloOpcode::kNegate, matrix_shape); in TEST_F()
476 ShapeInference::InferUnaryOpShape(unop, operand_shape)); in UnaryOp()