Searched refs:TransposeIsBitcast (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bitcast_decomposer_test.cc | 193 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose1->operand(0)->shape(), in TEST_P()
198 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose2->operand(0)->shape(), in TEST_P()
|
| D | bitcast_decomposer.cc | 113 CHECK(ShapeUtil::TransposeIsBitcast(input->shape(), shape, perm)) in Run()
|
| D | layout_normalization.cc | 303 if (!ShapeUtil::TransposeIsBitcast(s, operand_s, hlo->dimensions())) { in HandleTranspose()
|
| D | layout_assignment_test.cc | 556 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose->operand(0)->shape(), in TEST_F()
574 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose->operand(0)->shape(), in TEST_F()
590 EXPECT_DEATH(ShapeUtil::TransposeIsBitcast(hlo->operand(0)->shape(), in TEST_F()
|
| D | algebraic_simplifier.cc | 259 bool TransposeIsBitcast(const HloInstruction* transpose) { in TransposeIsBitcast() function
262 return ShapeUtil::TransposeIsBitcast(operand->shape(), transpose->shape(), in TransposeIsBitcast()
6122 if (options_.is_layout_sensitive() && TransposeIsBitcast(transpose)) { in HandleTranspose()
|
| D | hlo_instruction.cc | 2782 ShapeUtil::TransposeIsBitcast(operand(0)->shape(), shape(), in IsEffectiveBitcast()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 75 !ShapeUtil::TransposeIsBitcast(instr.operand(0)->shape(), in IsPhysicallyTransposing()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 571 static bool TransposeIsBitcast(const Shape& input_shape,
|
| D | shape_util.cc | 1187 CHECK(TransposeIsBitcast(shape, new_shape, permutation)) in PermuteDimensions()
1304 /* static */ bool ShapeUtil::TransposeIsBitcast( in TransposeIsBitcast() function in xla::ShapeUtil
|
| D | shape_util_test.cc | 742 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast( in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 322 ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) { in SourceIndexOfTranspose()
|