Searched refs:CompatibleIgnoringElementType (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 223 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
240 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
257 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
258 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
270 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
271 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
|
| D | shape_util.h | 287 static bool CompatibleIgnoringElementType(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 501 /* static */ bool ShapeUtil::CompatibleIgnoringElementType(const Shape& lhs, in CompatibleIgnoringElementType() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | kernel_tiling.cc | 82 if (!ShapeUtil::CompatibleIgnoringElementType(a, b)) { in FindTranspose021()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.cc | 1311 if (!ShapeUtil::CompatibleIgnoringElementType(operand_shape, out_shape)) { in CheckElementwiseInstruction()
|
| D | hlo_instruction.cc | 1225 if (sharding_ != nullptr && ShapeUtil::CompatibleIgnoringElementType( in SetupDerivedInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3481 TF_RET_CHECK(ShapeUtil::CompatibleIgnoringElementType( in AreFusedReductionOutputsConsistent()
|