Searched refs:ReshapeIsBitcast (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.h | 39 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const { in ReshapeIsBitcast() function
44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape); in ReshapeIsBitcast()
|
| D | layout_assignment_test.cc | 645 ShapeUtil::ReshapeIsBitcast(hlo->operand(0)->shape(), hlo->shape()), in TEST_F()
|
| D | algebraic_simplifier.cc | 199 if (options.ReshapeIsBitcast(operand->shape(), instr->shape())) { in BitcastingOperandOfReshapeOrCopyChainHelper()
790 ShapeUtil::ReshapeIsBitcast(copy->operand(0)->shape(), copy->shape())) { in HandleCopy()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 686 ShapeUtil::ReshapeIsBitcast( in TEST()
697 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast( in TEST()
765 EXPECT_FALSE(ShapeUtil::ReshapeIsBitcast( in TEST()
778 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
785 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
795 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
804 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
|
| D | shape_util.cc | 1127 /* static */ bool ShapeUtil::ReshapeIsBitcast(const Shape& input_shape, in ReshapeIsBitcast() function in xla::ShapeUtil
1429 CHECK(ReshapeIsBitcast(input_shape, output_shape_with_layout)) in AlignLayouts()
|
| D | shape_util.h | 621 static bool ReshapeIsBitcast(const Shape& input_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 133 ShapeUtil::ReshapeIsBitcast(a, b); in LinearValidOnShape()
179 ShapeUtil::ReshapeIsBitcast(input_shape, output_shape)) { in SourceIndexOfReshape()
229 if (ShapeUtil::ReshapeIsBitcast(operand_shape, shape)) { in SourceIndexOfBitcast()
|