Searched refs:CompatibleIgnoringFpPrecision (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 143 TEST(ShapeUtilTest, CompatibleIgnoringFpPrecision) { in TEST() argument
146 ASSERT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
152 ASSERT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
214 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
231 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(tuple1, tuple2)); in TEST()
259 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
260 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
268 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape1, shape2)); in TEST()
269 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringFpPrecision(shape2, shape1)); in TEST()
|
| D | shape_util.h | 291 static bool CompatibleIgnoringFpPrecision(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 506 /* 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()
884 if (ShapeUtil::CompatibleIgnoringFpPrecision(lhs, rhs)) { in InferElementwiseBinaryOpShape()
1084 if (ShapeUtil::CompatibleIgnoringFpPrecision(*arg_shapes[i], *arg_shape)) { in InferMapShape()
2145 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2153 if (!ShapeUtil::CompatibleIgnoringFpPrecision(operand_element_shape, in InferSelectAndScatterShape()
2172 if (!ShapeUtil::CompatibleIgnoringFpPrecision(source_shape, in InferSelectAndScatterShape()
2749 if (((ShapeUtil::CompatibleIgnoringFpPrecision(min, operand) || in InferClampShape()
2751 (ShapeUtil::CompatibleIgnoringFpPrecision(max, operand) || in InferClampShape()
[all …]
|
| D | hlo_verifier.h | 129 : ShapeUtil::CompatibleIgnoringFpPrecision(a, b); in ShapesSameIgnoringFpPrecision()
|
| D | hlo_verifier.cc | 367 if (!ShapeUtil::CompatibleIgnoringFpPrecision(expected_scalar_shape, in HandleSort()
888 return ShapeUtil::CompatibleIgnoringFpPrecision(instruction->shape(), in CheckShape()
|
| D | hlo_instruction.cc | 1892 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceUseWith()
1920 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringFpPrecision(old_operand->shape(), in ReplaceOperandWith()
1950 ShapeUtil::CompatibleIgnoringFpPrecision(shape(), new_producer->shape())) in ReplaceAllUsesWith()
|