Home
last modified time | relevance | path

Searched refs:InferBinaryOpShape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc268 return ShapeInference::InferBinaryOpShape(HloOpcode::kComplex, lhs, rhs, in TEST_F()
1362 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
1370 auto inferred_status = ShapeInference::InferBinaryOpShape( in TEST_F()
1832 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {1}); in TEST_F()
1837 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec8, {0}); in TEST_F()
1841 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {0}); in TEST_F()
1846 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, mat, vec16, {1}); in TEST_F()
1857 auto inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1862 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
1867 inferred_status_match = ShapeInference::InferBinaryOpShape( in TEST_F()
[all …]
Dshape_inference.h56 static StatusOr<Shape> InferBinaryOpShape(
59 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
Ddynamic_window_utils.cc54 ShapeInference::InferBinaryOpShape(opcode, x.get(), y.get()).ValueOrDie(); in BinaryOp()
Dhlo_creation_utils.cc51 ShapeInference::InferBinaryOpShape(opcode, lhs, rhs)); in MakeBinaryHlo()
63 ShapeInference::InferBinaryOpShape(HloOpcode::kCompare, lhs, rhs)); in MakeCompareHlo()
Dshape_inference.cc951 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
953 return InferBinaryOpShape(opcode, lhs->shape(), rhs->shape(), in InferBinaryOpShape()
957 /* static */ StatusOr<Shape> ShapeInference::InferBinaryOpShape( in InferBinaryOpShape() function in xla::ShapeInference
Dhlo_verifier.cc1257 instruction, ShapeInference::InferBinaryOpShape(instruction->opcode(), in CheckBinaryShape()
Dhlo_parser.cc1147 return ShapeInference::InferBinaryOpShape(opcode, operands[0], in ParseInstructionRhs()
1712 return ShapeInference::InferBinaryOpShape(opcode, operands[0], in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc633 Shape shape, ShapeInference::InferBinaryOpShape( in BinaryOp()