Searched refs:InferTernaryOpShape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 115 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 …]
|
| D | shape_inference.h | 65 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode, const Shape& lhs,
68 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode,
|
| D | hlo_creation_utils.cc | 405 ShapeInference::InferTernaryOpShape(select_op_code, pred, in MakeSelectHlo()
|
| D | shape_inference.cc | 1042 /* 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
|
| D | hlo_verifier.cc | 1264 ShapeInference::InferTernaryOpShape( in CheckTernaryShape()
|
| D | hlo_parser.cc | 1167 return ShapeInference::InferTernaryOpShape( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 776 StatusOr<const Shape> status_or_shape = ShapeInference::InferTernaryOpShape( in TernaryOp()
|