Searched refs:InferUnaryOpShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 49 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
51 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
|
| D | hlo_creation_utils.cc | 41 ShapeInference::InferUnaryOpShape(opcode, operand)); in MakeUnaryHlo()
|
| D | shape_inference.cc | 249 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape() function in xla::ShapeInference
251 return InferUnaryOpShape(opcode, operand->shape()); in InferUnaryOpShape()
254 /* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape( in InferUnaryOpShape() function in xla::ShapeInference
|
| D | hlo_verifier.cc | 1251 ShapeInference::InferUnaryOpShape(instruction->opcode(), in CheckUnaryShape()
|
| D | hlo_parser.cc | 1115 return ShapeInference::InferUnaryOpShape(opcode, operands[0]); in ParseInstructionRhs()
2548 return ShapeInference::InferUnaryOpShape(opcode, operands[0]); in ParseInstructionRhs()
|
| D | shape_inference_test.cc | 108 ShapeInference::InferUnaryOpShape(HloOpcode::kNegate, matrix_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 620 Shape shape, ShapeInference::InferUnaryOpShape(unop, *operand_shape)); in UnaryOp()
|