/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_rematerialization_test_utils.h | 68 auto concat_1 = builder.AddInstruction(HloInstruction::CreateConcatenate( 75 auto concat_2 = builder.AddInstruction(HloInstruction::CreateConcatenate( 117 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate(
|
D | all_to_all_decomposer.cc | 126 HloInstruction::CreateConcatenate(all_to_all->shape(), gtes, split_dim)); in ExpandInstruction()
|
D | hlo_subcomputation_unification_test.cc | 141 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
D | hlo_constant_folding_test.cc | 137 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
D | hlo_rematerialization_test.cc | 352 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F() 509 auto concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
|
D | hlo_creation_utils.cc | 249 HloInstruction::CreateConcatenate(concat_shape, operands, dimension)); in MakeConcatHlo()
|
D | algebraic_simplifier_test.cc | 2112 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()
|
D | hlo_instruction.h | 796 static std::unique_ptr<HloInstruction> CreateConcatenate(
|
D | buffer_assignment_test.cc | 1775 HloInstruction::CreateConcatenate(shape_5x4, {dot_ab, dot_bc}, 0)); in TEST_F() 1877 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
|
D | layout_assignment_test.cc | 554 HloInstruction::CreateConcatenate(bshape, {param0, param1}, 1)); in TEST_F()
|
D | dynamic_dimension_inference_test.cc | 1051 auto* concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
|
D | hlo_evaluator_test.cc | 564 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F() 586 b.AddInstruction(HloInstruction::CreateConcatenate(shape, operands, 0)); in TEST_F()
|
D | hlo_instruction.cc | 234 CreateConcatenate(shape, all_operands(), proto.dimensions(0)); in CreateFromProto() 1254 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConcatenate( in CreateConcatenate() function in xla::HloInstruction
|
D | hlo_parser.cc | 1781 instruction = builder->AddInstruction(HloInstruction::CreateConcatenate( in ParseInstructionRhs()
|
/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()
|
D | gpu_fusion_test.cc | 80 HloInstruction::CreateConcatenate(concat_shape, slice_params, 1)); 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/tests/ |
D | custom_call_test.cc | 131 HloInstruction::CreateConcatenate(ShapeUtil::MakeShape(F32, {2, 2, 2}), in XLA_TEST_F()
|
D | cpu_gpu_fusion_test.cc | 212 auto concat5 = builder.AddInstruction(HloInstruction::CreateConcatenate( in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | gather_scatter_handler.cc | 104 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()
|
D | dot_handler.cc | 888 dot_lhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase() 911 dot_rhs = body_b.AddInstruction(HloInstruction::CreateConcatenate( in PartitionBaseCase()
|
D | spmd_partitioner_util.cc | 893 HloInstruction::CreateConcatenate(concat_shape, concat_pieces, dim)); in ExchangeHalo()
|