Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc182 TEST(ShapeUtilTest, EqualIgnoringElementType) { in TEST() argument
183 EXPECT_TRUE(ShapeUtil::EqualIgnoringElementType( in TEST()
186 EXPECT_TRUE(ShapeUtil::EqualIgnoringElementType( in TEST()
189 EXPECT_TRUE(ShapeUtil::EqualIgnoringElementType( in TEST()
195 EXPECT_FALSE(ShapeUtil::EqualIgnoringElementType( in TEST()
198 EXPECT_FALSE(ShapeUtil::EqualIgnoringElementType( in TEST()
Dshape_util.h329 static bool EqualIgnoringElementType(const Shape& lhs, const Shape& rhs);
Dshape_util.cc160 /* static */ bool ShapeUtil::EqualIgnoringElementType(const Shape& lhs, in EqualIgnoringElementType() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.cc99 if (!ShapeUtil::EqualIgnoringElementType(shape_a, shape_b)) { in FindAndSortFusionCandidates()
Dgpu_fusible.cc190 return ShapeUtil::EqualIgnoringElementType(get_loop_shape(instr_1), in ShapesCompatibleForMultiOutputFusion()
Dir_emitter_unnested.cc513 if (!ShapeUtil::EqualIgnoringElementType(*first_slice_operand_shape, in GetConsistentInputShapeForRootSlices()