/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | conv_canonicalization.cc | 75 HloInstruction::CreateTranspose(new_input_shape, input, in Run() 97 HloInstruction::CreateTranspose(new_kernel_shape, kernel, in Run() 140 hlo, HloInstruction::CreateTranspose( in Run()
|
D | cpu_instruction_fusion_test.cc | 491 HloInstruction::CreateTranspose(result_shape, exp1, {1, 0})); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dot_decomposer.cc | 124 computation->AddInstruction(HloInstruction::CreateTranspose( in DecomposeBatchDot() 131 computation->AddInstruction(HloInstruction::CreateTranspose( in DecomposeBatchDot() 207 computation->AddInstruction(HloInstruction::CreateTranspose( in CanonicalizeDot() 251 computation->AddInstruction(HloInstruction::CreateTranspose( in CanonicalizeDot()
|
D | bfloat16_propagation_test.cc | 117 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 178 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 279 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 328 HloInstruction* lhs = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 541 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 654 HloInstruction::CreateTranspose(shape, body_param, {0, 1})); in TEST_F() 659 builder_f.AddInstruction(HloInstruction::CreateTranspose(shape, a_f, {0, 1})); in TEST_F() 737 HloInstruction::CreateTranspose(shape, body_dot2, {0, 1})); in TEST_F() 944 builder.AddInstruction(HloInstruction::CreateTranspose(shape, a, {0, 1})); in TEST_F() 946 builder.AddInstruction(HloInstruction::CreateTranspose(shape, b, {0, 1})); in TEST_F() [all …]
|
D | transpose_folding_test.cc | 228 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 284 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 346 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 413 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
|
D | reshape_mover.cc | 156 return computation->AddInstruction(HloInstruction::CreateTranspose( in UpdateOperand() 248 HloInstruction::CreateTranspose(instruction->shape(), new_elementwise, in PerformSinkReshapeOrTranspose()
|
D | layout_assignment_test.cc | 401 HloInstruction::CreateTranspose(bshape, log, {1, 0})); in TEST_F() 439 HloInstruction::CreateTranspose(cshape, broadcast, {2, 1, 0})); in TEST_F() 482 HloInstruction::CreateTranspose(f32_43, broadcast, {1, 0})); in TEST_F() 589 auto transpose = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 607 auto transpose = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 627 HloInstruction::CreateTranspose(input_shape, param, {0, 2, 1})); in TEST_F() 1332 b.AddInstruction(HloInstruction::CreateTranspose(ashape, param0, {1, 0})); in TEST_F()
|
D | hlo_instruction_test.cc | 1148 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 1189 builder.AddInstruction(HloInstruction::CreateTranspose(s, y, {1, 0})); in TEST_F() 1240 HloInstruction::CreateTranspose(data_shape, b, {1, 0})); in TEST_F() 1321 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 1486 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 1528 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 1584 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
|
D | hlo_computation_test.cc | 549 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 584 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F() 620 builder.AddInstruction(HloInstruction::CreateTranspose(s2t, y, {1, 0})); in TEST_F()
|
D | hlo_constant_folding_test.cc | 188 HloInstruction::CreateTranspose(shape, literal_instruction, permutation)); in TEST_F()
|
D | instruction_fusion_test.cc | 149 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
|
D | hlo_creation_utils.cc | 104 HloInstruction::CreateTranspose(transpose_shape, operand, dimensions)); in MakeTransposeHlo()
|
D | algebraic_simplifier_test.cc | 1982 HloInstruction::CreateTranspose(rng0->shape(), rng0, {1, 0})); in TEST_F() 2110 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 2140 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 2224 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 2227 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 3549 builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F() 4402 HloInstruction::CreateTranspose(lhs_shape, lhs, {1, 0})); in TEST_P() 4408 HloInstruction::CreateTranspose(rhs_shape, rhs, {1, 0})); in TEST_P()
|
D | dynamic_dimension_inference_test.cc | 351 HloInstruction::CreateTranspose(output_shape, a_param, {2, 1, 0})); in TEST_F()
|
D | hlo_instruction.h | 691 static std::unique_ptr<HloInstruction> CreateTranspose(
|
D | algebraic_simplifier.cc | 1815 dot, HloInstruction::CreateTranspose(dot->shape(), new_dot, {1, 0})); in HandleDot() 2058 HloInstruction::CreateTranspose(broadcast->shape(), operand, dims)); in HandleBroadcast() 3365 transpose, HloInstruction::CreateTranspose( in HandleTranspose()
|
D | heap_simulator_test.cc | 184 HloInstruction::CreateTranspose(r2f32, matrix, {0, 1})); in TEST_F()
|
D | hlo_instruction.cc | 272 CreateTranspose(shape, operands(0), in CreateFromProto() 1186 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateTranspose( in CreateTranspose() function in xla::HloInstruction
|
D | hlo_evaluator_test.cc | 428 HloInstruction::CreateTranspose(shape, literal_instruction, permutation)); in TEST_F()
|
D | hlo_parser.cc | 1311 HloInstruction::CreateTranspose(shape, operands[0], *dimensions)); in ParseInstructionRhs()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_eigen_dot_operation_test.cc | 87 HloInstruction::CreateTranspose(param_shape, lhs, {1, 0})); in TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | fusion_test.cc | 209 auto reshape3 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() 441 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() 456 auto reshape1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F() 581 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | instruction_fusion_test.cc | 100 HloInstruction::CreateTranspose(ShapeUtil::MakeShape(S32, {}), exp1, {})); in TEST_F() 134 auto transpose2 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
|