Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc113 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
120 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
128 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
135 auto inferred_status_error1 = ShapeInference::InferTernaryOpShape( in TEST_F()
141 auto inferred_status_error2 = ShapeInference::InferTernaryOpShape( in TEST_F()
147 auto inferred_status_error3 = ShapeInference::InferTernaryOpShape( in TEST_F()
155 auto inferred_status_error4 = ShapeInference::InferTernaryOpShape( in TEST_F()
165 auto inferred_status = ShapeInference::InferTernaryOpShape( in TEST_F()
173 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, f32_, f32_); in TEST_F()
179 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.cc282 ShapeInference::InferTernaryOpShape( in MakeSelectHlo()
Dshape_inference.cc998 /* static */ StatusOr<Shape> ShapeInference::InferTernaryOpShape( in InferTernaryOpShape() function in xla::ShapeInference
1001 return InferTernaryOpShape(opcode, lhs->shape(), rhs->shape(), ehs->shape()); in InferTernaryOpShape()
1004 /* static */ StatusOr<Shape> ShapeInference::InferTernaryOpShape( in InferTernaryOpShape() function in xla::ShapeInference
Dhlo_verifier.cc931 ShapeInference::InferTernaryOpShape( in CheckTernaryShape()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc561 Shape shape, ShapeInference::InferTernaryOpShape(triop, lhs_shape, in TernaryOp()