Searched refs:CompatibleIgnoringElementType (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 247 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
264 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
281 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
282 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
294 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
295 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
|
| D | shape_util.h | 314 static bool CompatibleIgnoringElementType(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 658 /* static */ bool ShapeUtil::CompatibleIgnoringElementType(const Shape& lhs, in CompatibleIgnoringElementType() function in xla::ShapeUtil
1689 if (!CompatibleIgnoringElementType(a, b)) { in FindTranspose021()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | slice_sinker.cc | 61 return ShapeUtil::CompatibleIgnoringElementType( in IsElementwiseOperationOnSimilarSlices()
|
| D | space_to_batch_converter.cc | 1018 if (ShapeUtil::CompatibleIgnoringElementType(previous_broadcast->shape(), in PropagateOnBroadcast()
1245 if (ShapeUtil::CompatibleIgnoringElementType( in Propagate()
|
| D | sharding_propagation.cc | 1395 if (ShapeUtil::CompatibleIgnoringElementType(instruction.shape(), in GetShardingFromUser()
|
| D | hlo_verifier.cc | 1748 if (!ShapeUtil::CompatibleIgnoringElementType(operand_shape, out_shape)) { in CheckElementwiseInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 607 return ShapeUtil::CompatibleIgnoringElementType( in IsFusedReductionOutputConsistent()
639 if (!(ShapeUtil::CompatibleIgnoringElementType(first_reduce_operand_shape, in IsFusedReductionOutputConsistent()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 2897 return ShapeUtil::CompatibleIgnoringElementType( in FindInputNodesIfOnlyDependOnSmallOperands()
3100 return ShapeUtil::CompatibleIgnoringElementType( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|