| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_noalias_test.cc | 43 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
45 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | variadic_op_splitter_test.cc | 55 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
70 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_noalias_test.cc | 51 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
54 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
| D | cpu_fusion_test.cc | 155 HloInstruction::CreateConcatenate(cshape, {ceil, ceil}, /*dimension=*/0)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_rematerialization_test.cc | 69 auto concat_1 = builder.AddInstruction(HloInstruction::CreateConcatenate( in MakeRematerializableComputation()
76 auto concat_2 = builder.AddInstruction(HloInstruction::CreateConcatenate( in MakeRematerializableComputation()
118 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in MakeRematerializableWhileComputation()
430 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
589 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
|
| D | hlo_subcomputation_unification_test.cc | 141 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
| D | hlo_constant_folding_test.cc | 135 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
| D | dot_decomposer.cc | 152 HloInstruction::CreateConcatenate(concat_shape_r3, output_slices, 0)); in DecomposeBatchDot()
|
| D | convolution_group_converter.cc | 529 auto new_conv = HloInstruction::CreateConcatenate( in HandleConvolution()
|
| D | hlo_creation_utils.cc | 214 HloInstruction::CreateConcatenate(concat_shape, operands, dimension)); in MakeConcatHlo()
|
| D | algebraic_simplifier_test.cc | 1659 HloInstruction::CreateConcatenate(param0->shape(), {param0}, 0)); in TEST_F()
1688 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
1722 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
1770 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
1797 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
1859 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
4482 HloInstruction::CreateConcatenate(rhs_shape, {rhs0, rhs1, rhs2}, 0)); in TEST_P()
4541 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
|
| D | hlo_instruction.h | 597 static std::unique_ptr<HloInstruction> CreateConcatenate(
|
| D | buffer_assignment_test.cc | 1603 HloInstruction::CreateConcatenate(shape_5x4, {dot_ab, dot_bc}, 0)); in TEST_F()
1705 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
|
| D | hlo_evaluator_test.cc | 487 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F()
509 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F()
|
| D | layout_assignment_test.cc | 554 HloInstruction::CreateConcatenate(bshape, {param0, param1}, 1)); in TEST_F()
|
| D | hlo_instruction.cc | 212 CreateConcatenate(shape, all_operands(), proto.dimensions(0)); in CreateFromProto()
1008 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConcatenate( in CreateConcatenate() function in xla::HloInstruction
|
| D | hlo_parser.cc | 1175 instruction = builder->AddInstruction(HloInstruction::CreateConcatenate( in ParseInstructionRhs()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 131 HloInstruction::CreateConcatenate(ShapeUtil::MakeShape(F32, {2, 2, 2}), in XLA_TEST_F()
|
| D | fusion_test.cc | 213 auto concat5 = builder.AddInstruction(HloInstruction::CreateConcatenate( in XLA_TEST_F()
|