Searched refs:ReshapeIsBitcast (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 646 ShapeUtil::ReshapeIsBitcast( in TEST()
657 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast( in TEST()
663 EXPECT_FALSE(ShapeUtil::ReshapeIsBitcast( in TEST()
676 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
683 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
695 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
|
| D | shape_util.h | 531 static bool ReshapeIsBitcast(const Shape& input_shape,
|
| D | shape_util.cc | 1107 /* static */ bool ShapeUtil::ReshapeIsBitcast(const Shape& input_shape, in ReshapeIsBitcast() function in xla::ShapeUtil
1399 CHECK(ReshapeIsBitcast(input_shape, output_shape_with_layout)); in AlignLayouts()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion.cc | 120 return ShapeUtil::ReshapeIsBitcast( in NotWorthHoistingIndividually()
|
| D | layout_assignment.cc | 930 if (ShapeUtil::ReshapeIsBitcast(operand_shape, output_shape_with_layout)) { in ChooseOperandLayoutFromOutputLayout()
935 if (ShapeUtil::ReshapeIsBitcast(operand_shape, in ChooseOperandLayoutFromOutputLayout()
1002 if (ShapeUtil::ReshapeIsBitcast(output_shape, operand_shape_with_layout)) { in ChooseOutputLayoutFromOperandLayout()
1007 if (ShapeUtil::ReshapeIsBitcast(output_shape, in ChooseOutputLayoutFromOperandLayout()
|
| D | algebraic_simplifier.cc | 79 bool ReshapeIsBitcast( in ReshapeIsBitcast() function
87 return ShapeUtil::ReshapeIsBitcast(operand->shape(), reshape->shape()) && in ReshapeIsBitcast()
1544 ReshapeIsBitcast(reshape, valid_bitcast_callback_)) { in HandleReshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 164 ShapeUtil::ReshapeIsBitcast(input_shape, output_shape)) { in SourceIndexOfReshape()
|