Home
last modified time | relevance | path

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

12

/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_reduce_contiguous.cc53 computation.AddInstruction(HloInstruction::CreateConcatenate( in ReplaceWithContiguousAllReduce()
Dall_to_all_decomposer.cc127 HloInstruction::CreateConcatenate(all_to_all->shape(), gtes, split_dim)); in ExpandInstruction()
Ddynamic_dimension_simplifier.cc46 auto new_concat = parent->AddInstruction(HloInstruction::CreateConcatenate( in ConcatForwarding()
Dhlo_subcomputation_unification_test.cc141 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dselect_and_scatter_expander.cc177 auto* indices = computation->AddInstruction(HloInstruction::CreateConcatenate( in ExpandInstruction()
Ddot_merger.cc176 diff_op_a->AddInstruction(HloInstruction::CreateConcatenate( in TryMergeSameOperand()
Dhlo_constant_folding_test.cc138 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()
Dlayout_normalization.cc131 HloInstruction::CreateConcatenate(normalized_shape, normalized_inputs, in HandleConcatenate()
Dhlo_creation_utils.cc289 HloInstruction::CreateConcatenate(concat_shape, operands, dimension), in MakeConcatHlo()
Dwhile_loop_concat_code_motion.cc154 return comp->AddInstruction(HloInstruction::CreateConcatenate( in CreateConcat()
Dalgebraic_simplifier_test.cc2527 HloInstruction::CreateConcatenate(param0->shape(), {param0}, 0)); in TEST_F()
2630 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2664 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2712 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2739 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
2805 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
5987 HloInstruction::CreateConcatenate(rhs_shape, {rhs0, rhs1, rhs2}, 0)); in TEST_P()
6046 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_P()
Ddynamic_dimension_inference_test.cc1058 auto* concat = builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dbuffer_assignment_test.cc1811 HloInstruction::CreateConcatenate(shape_5x4, {dot_ab, dot_bc}, 0)); in TEST_F()
1913 HloInstruction::CreateConcatenate(concat_shape, {rev, neg}, 0)); in TEST_F()
Dlayout_assignment_test.cc517 HloInstruction::CreateConcatenate(bshape, {param0, param1}, 1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_noalias_test.cc42 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
44 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.cc50 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
53 builder.AddInstruction(HloInstruction::CreateConcatenate( in TEST_F()
Dcpu_fusion_test.cc154 HloInstruction::CreateConcatenate(cshape, {ceil, ceil}, /*dimension=*/0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc156 HloInstruction::CreateConcatenate(ShapeUtil::MakeShape(F32, {2, 2, 2}), in XLA_TEST_F()
Dcpu_gpu_fusion_test.cc210 auto concat5 = builder.AddInstruction(HloInstruction::CreateConcatenate( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc137 min_indices[0] = b->AddInstruction(HloInstruction::CreateConcatenate( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
141 max_indices[0] = b->AddInstruction(HloInstruction::CreateConcatenate( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
578 adjusted_indices = b->AddInstruction(HloInstruction::CreateConcatenate( in PartitionGatherIndexParallelDimensions()
Dcustom_call_handler.cc322 return b_.AddInstruction(HloInstruction::CreateConcatenate( in HandleCustomCallSPMDInternal_RotateRight()

12