Home
last modified time | relevance | path

Searched refs:InferTernaryOpShape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc115 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
123 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
133 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
140 auto inferred_status_error1 = ShapeInference::InferTernaryOpShape( in TEST_F()
146 auto inferred_status_error2 = ShapeInference::InferTernaryOpShape( in TEST_F()
152 auto inferred_status_error3 = ShapeInference::InferTernaryOpShape( in TEST_F()
161 auto inferred_status_error4 = ShapeInference::InferTernaryOpShape( in TEST_F()
171 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
179 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, f32_, f32_); in TEST_F()
185 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
[all …]
Dshape_inference.h65 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode, const Shape& lhs,
68 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode,
Dhlo_creation_utils.cc405 ShapeInference::InferTernaryOpShape(select_op_code, pred, in MakeSelectHlo()
Dshape_inference.cc1042 /* static */ StatusOr<Shape> ShapeInference::InferTernaryOpShape( in InferTernaryOpShape() function in xla::ShapeInference
1045 return InferTernaryOpShape(opcode, lhs->shape(), rhs->shape(), ehs->shape()); in InferTernaryOpShape()
1048 /* static */ StatusOr<Shape> ShapeInference::InferTernaryOpShape( in InferTernaryOpShape() function in xla::ShapeInference
Dhlo_verifier.cc1264 ShapeInference::InferTernaryOpShape( in CheckTernaryShape()
Dhlo_parser.cc1167 return ShapeInference::InferTernaryOpShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc776 StatusOr<const Shape> status_or_shape = ShapeInference::InferTernaryOpShape( in TernaryOp()