Home
last modified time | relevance | path

Searched refs:ReshapeIsBitcast (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.h39 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const { in ReshapeIsBitcast() function
44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape); in ReshapeIsBitcast()
Dlayout_assignment_test.cc648 ShapeUtil::ReshapeIsBitcast(hlo->operand(0)->shape(), hlo->shape()), in TEST_F()
Dalgebraic_simplifier.cc133 if (options.ReshapeIsBitcast(operand->shape(), instr->shape())) { in BitcastingOperandOfReshapeOrCopyChainHelper()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc664 ShapeUtil::ReshapeIsBitcast( in TEST()
675 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast( in TEST()
734 EXPECT_FALSE(ShapeUtil::ReshapeIsBitcast( in TEST()
747 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
754 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
764 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
773 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value())); in TEST()
Dshape_util.cc1100 /* static */ bool ShapeUtil::ReshapeIsBitcast(const Shape& input_shape, in ReshapeIsBitcast() function in xla::ShapeUtil
1402 CHECK(ReshapeIsBitcast(input_shape, output_shape_with_layout)) in AlignLayouts()
Dshape_util.h594 static bool ReshapeIsBitcast(const Shape& input_shape,
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc127 ShapeUtil::ReshapeIsBitcast(a, b); in LinearValidOnShape()
174 ShapeUtil::ReshapeIsBitcast(input_shape, output_shape)) { in SourceIndexOfReshape()
224 if (ShapeUtil::ReshapeIsBitcast(operand_shape, shape)) { in SourceIndexOfBitcast()