Searched refs:CompatibleIgnoringFpPrecision (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 144 TEST(ShapeUtilTest, CompatibleIgnoringFpPrecision) { in TEST() argument
147 ASSERT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
153 ASSERT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
236 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
253 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
281 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
282 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
290 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
291 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
|
| D | shape_util.h | 290 static bool CompatibleIgnoringFpPrecision(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 556 /* static */ bool ShapeUtil::CompatibleIgnoringFpPrecision(const Shape& lhs, in CompatibleIgnoringFpPrecision() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 120 if (!ShapeUtil::CompatibleIgnoringFpPrecision(*accumulator_subshapes[i], in VerifyReducerShape()
131 if (!ShapeUtil::CompatibleIgnoringFpPrecision( in VerifyReducerShape()
145 if (!ShapeUtil::CompatibleIgnoringFpPrecision( in VerifyReducerShape()
899 if (ShapeUtil::CompatibleIgnoringFpPrecision(lhs, rhs)) { in InferElementwiseBinaryOpShape()
1108 if (ShapeUtil::CompatibleIgnoringFpPrecision(*arg_shapes[i], *arg_shape)) { in InferMapShape()
2177 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2185 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2204 if (!ShapeUtil::CompatibleIgnoringFpPrecision(source_shape, in InferSelectAndScatterShape()
2974 if (!ShapeUtil::CompatibleIgnoringFpPrecision(min, operand) || in InferClampShape()
2975 !ShapeUtil::CompatibleIgnoringFpPrecision(max, operand)) { in InferClampShape()
[all …]
|
| D | hlo_verifier.h | 150 return ShapeUtil::CompatibleIgnoringFpPrecision(a, b);
|
| D | hlo_verifier.cc | 467 if (!ShapeUtil::CompatibleIgnoringFpPrecision(expected_scalar_shape, in HandleSort()
1081 return ShapeUtil::CompatibleIgnoringFpPrecision(instruction->shape(), in CheckShape()
|
| D | hlo_instruction.cc | 2032 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
2061 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(old_operand->shape(), in ReplaceOperandWith()
2091 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceAllUsesWith()
|
| D | hlo_evaluator.cc | 2398 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(reduce->shape(), in HandleReduce()
|