Home
last modified time | relevance | path

Searched refs:CreateTranspose (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc75 HloInstruction::CreateTranspose(new_input_shape, input, in Run()
97 HloInstruction::CreateTranspose(new_kernel_shape, kernel, in Run()
140 hlo, HloInstruction::CreateTranspose( in Run()
Dcpu_instruction_fusion_test.cc491 HloInstruction::CreateTranspose(result_shape, exp1, {1, 0})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc124 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()
Dbfloat16_propagation_test.cc117 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 …]
Dtranspose_folding_test.cc228 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()
Dreshape_mover.cc156 return computation->AddInstruction(HloInstruction::CreateTranspose( in UpdateOperand()
248 HloInstruction::CreateTranspose(instruction->shape(), new_elementwise, in PerformSinkReshapeOrTranspose()
Dlayout_assignment_test.cc401 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()
Dhlo_instruction_test.cc1148 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()
Dhlo_computation_test.cc549 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()
Dhlo_constant_folding_test.cc188 HloInstruction::CreateTranspose(shape, literal_instruction, permutation)); in TEST_F()
Dinstruction_fusion_test.cc149 auto transpose1 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()
Dhlo_creation_utils.cc104 HloInstruction::CreateTranspose(transpose_shape, operand, dimensions)); in MakeTransposeHlo()
Dalgebraic_simplifier_test.cc1982 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()
Ddynamic_dimension_inference_test.cc351 HloInstruction::CreateTranspose(output_shape, a_param, {2, 1, 0})); in TEST_F()
Dhlo_instruction.h691 static std::unique_ptr<HloInstruction> CreateTranspose(
Dalgebraic_simplifier.cc1815 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()
Dheap_simulator_test.cc184 HloInstruction::CreateTranspose(r2f32, matrix, {0, 1})); in TEST_F()
Dhlo_instruction.cc272 CreateTranspose(shape, operands(0), in CreateFromProto()
1186 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateTranspose( in CreateTranspose() function in xla::HloInstruction
Dhlo_evaluator_test.cc428 HloInstruction::CreateTranspose(shape, literal_instruction, permutation)); in TEST_F()
Dhlo_parser.cc1311 HloInstruction::CreateTranspose(shape, operands[0], *dimensions)); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc87 HloInstruction::CreateTranspose(param_shape, lhs, {1, 0})); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc209 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/
Dinstruction_fusion_test.cc100 HloInstruction::CreateTranspose(ShapeUtil::MakeShape(S32, {}), exp1, {})); in TEST_F()
134 auto transpose2 = builder.AddInstruction(HloInstruction::CreateTranspose( in TEST_F()