Home
last modified time | relevance | path

Searched refs:CreateConcatenate (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test_utils.h68 auto concat_1 = builder.AddInstruction(HloInstruction::CreateConcatenate(
75 auto concat_2 = builder.AddInstruction(HloInstruction::CreateConcatenate(
117 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate(
Dall_to_all_decomposer.cc126 HloInstruction::CreateConcatenate(all_to_all->shape(), gtes, split_dim)); in ExpandInstruction()
Dhlo_subcomputation_unification_test.cc141 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dhlo_constant_folding_test.cc137 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dhlo_rematerialization_test.cc352 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
509 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
Dhlo_creation_utils.cc249 HloInstruction::CreateConcatenate(concat_shape, operands, dimension)); in MakeConcatHlo()
Dalgebraic_simplifier_test.cc2112 HloInstruction::CreateConcatenate(param0->shape(), {param0}, 0)); in TEST_F()
2215 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2249 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2297 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2324 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2390 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
5346 HloInstruction::CreateConcatenate(rhs_shape, {rhs0, rhs1, rhs2}, 0)); in TEST_P()
5405 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
Dhlo_instruction.h796 static std::unique_ptr<HloInstruction> CreateConcatenate(
Dbuffer_assignment_test.cc1775 HloInstruction::CreateConcatenate(shape_5x4, {dot_ab, dot_bc}, 0)); in TEST_F()
1877 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
Dlayout_assignment_test.cc554 HloInstruction::CreateConcatenate(bshape, {param0, param1}, 1)); in TEST_F()
Ddynamic_dimension_inference_test.cc1051 auto* concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dhlo_evaluator_test.cc564 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F()
586 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F()
Dhlo_instruction.cc234 CreateConcatenate(shape, all_operands(), proto.dimensions(0)); in CreateFromProto()
1254 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConcatenate( in CreateConcatenate() function in xla::HloInstruction
Dhlo_parser.cc1781 instruction = builder->AddInstruction(HloInstruction::CreateConcatenate( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_noalias_test.cc43 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
45 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dgpu_fusion_test.cc80 HloInstruction::CreateConcatenate(concat_shape, slice_params, 1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dvariadic_op_splitter_test.cc55 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
70 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc51 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
54 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dcpu_fusion_test.cc155 HloInstruction::CreateConcatenate(cshape, {ceil, ceil}, /*dimension=*/0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc131 HloInstruction::CreateConcatenate(ShapeUtil::MakeShape(F32, {2, 2, 2}), in XLA_TEST_F()
Dcpu_gpu_fusion_test.cc212 auto concat5 = builder.AddInstruction(HloInstruction::CreateConcatenate( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc104 min_indices[0] = b->AddInstruction(HloInstruction::CreateConcatenate( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
108 max_indices[0] = b->AddInstruction(HloInstruction::CreateConcatenate( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
392 adjusted_indices = b->AddInstruction(HloInstruction::CreateConcatenate( in PartitionIndexParallelDimensions()
Ddot_handler.cc888 dot_lhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase()
911 dot_rhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase()
Dspmd_partitioner_util.cc893 HloInstruction::CreateConcatenate(concat_shape, concat_pieces, dim)); in ExchangeHalo()