Searched refs:CompatibleIgnoringElementType (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 244 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
261 EXPECT_TRUE(ShapeUtil::CompatibleIgnoringElementType(tuple1, tuple2)); in TEST()
278 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
279 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
291 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape1, shape2)); in TEST()
292 EXPECT_FALSE(ShapeUtil::CompatibleIgnoringElementType(shape2, shape1)); in TEST()
|
| D | shape_util.h | 197 static bool CompatibleIgnoringElementType(const Shape& lhs, const Shape& rhs);
|
| D | shape_util.cc | 739 /* static */ bool ShapeUtil::CompatibleIgnoringElementType(const Shape& lhs, in CompatibleIgnoringElementType() function in xla::ShapeUtil
1694 if (!ShapeUtil::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 | 1428 if (ShapeUtil::CompatibleIgnoringElementType(previous_broadcast->shape(), in PropagateOnBroadcast()
1780 if (ShapeUtil::CompatibleIgnoringElementType( in Propagate()
|
| D | hlo_verifier.cc | 2320 if (!ShapeUtil::CompatibleIgnoringElementType(operand_shape, out_shape)) { in CheckElementwiseInstruction()
|
| D | sharding_propagation.cc | 1603 if (ShapeUtil::CompatibleIgnoringElementType(instruction.shape(), in GetShardingFromUser()
|
| D | shape_inference.cc | 1126 if (ShapeUtil::CompatibleIgnoringElementType(*arg_shapes[i], *arg_shape)) { in InferMapShape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bincount_op.cc | 84 xla::ShapeUtil::CompatibleIgnoringElementType(weights_shape, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 52 if (!ShapeUtil::CompatibleIgnoringElementType(operands_shapes[0], in GetOperandTypes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 471 return ShapeUtil::CompatibleIgnoringElementType( in IsFusedReductionOutputConsistent()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 3833 return ShapeUtil::CompatibleIgnoringElementType( in FindInputNodesIfOnlyDependOnSmallOperands()
4053 return ShapeUtil::CompatibleIgnoringElementType( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|