Searched refs:CompatibleIgnoringFpPrecision (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 137 TEST(ShapeUtilTest, CompatibleIgnoringFpPrecision) { in TEST() argument
140 ASSERT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
146 ASSERT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
235 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
252 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
280 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
281 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
289 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
290 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
|
| D | shape_util.h | 206 static bool CompatibleIgnoringFpPrecision(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 756 /* static */ bool ShapeUtil::CompatibleIgnoringFpPrecision(const Shape& lhs, in CompatibleIgnoringFpPrecision() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 123 if (!ShapeUtil::CompatibleIgnoringFpPrecision(*accumulator_subshapes[i], in VerifyReducerShape()
134 if (!ShapeUtil::CompatibleIgnoringFpPrecision( in VerifyReducerShape()
148 if (!ShapeUtil::CompatibleIgnoringFpPrecision( in VerifyReducerShape()
923 if (ShapeUtil::CompatibleIgnoringFpPrecision(lhs, rhs)) { in InferElementwiseBinaryOpShape()
2372 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2380 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2398 if (!ShapeUtil::CompatibleIgnoringFpPrecision(source_shape, in InferSelectAndScatterShape()
3258 if (!ShapeUtil::CompatibleIgnoringFpPrecision(min, operand) || in InferClampShape()
3259 !ShapeUtil::CompatibleIgnoringFpPrecision(max, operand)) { in InferClampShape()
3273 if (!ShapeUtil::CompatibleIgnoringFpPrecision(on_true, on_false)) { in InferSelectShape()
|
| D | hlo_verifier.h | 260 return ShapeUtil::CompatibleIgnoringFpPrecision(a, b);
|
| D | hlo_instruction.cc | 2634 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
2663 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
2690 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(old_operand->shape(), in ReplaceOperandWith()
2721 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUsesWith()
2741 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceAllUsesWith()
|
| D | hlo_verifier.cc | 926 if (!ShapeUtil::CompatibleIgnoringFpPrecision(expected_scalar_shape, in HandleSort()
1755 return ShapeUtil::CompatibleIgnoringFpPrecision(instruction->shape(), in CheckShape()
|
| D | hlo_evaluator.cc | 3723 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(reduce->shape(), in HandleReduce()
|