Home
last modified time | relevance | path

Searched refs:CreateSlice (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc72 auto slice_1 = builder.AddInstruction(HloInstruction::CreateSlice( in MakeRematerializableComputation()
81 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat_2, in MakeRematerializableComputation()
113 HloInstruction::CreateSlice(vec1_shape_, bcast, /*start_indices=*/{0}, in MakeRematerializableWhileComputation()
121 builder.AddInstruction(HloInstruction::CreateSlice(vec1_shape_, concat, in MakeRematerializableWhileComputation()
433 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
592 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_P()
Dbfloat16_propagation_test.cc582 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
587 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
639 builder_cond.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
644 builder_cond.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
715 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
720 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
811 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
816 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
840 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
845 HloInstruction::CreateSlice(ShapeUtil::MakeShape(F32, {1, 1}), in TEST_F()
Ddynamic_index_splitter.cc75 auto slice = parent->AddInstruction(HloInstruction::CreateSlice( in Run()
Dhlo_creation_utils.cc78 return computation->AddInstruction(HloInstruction::CreateSlice( in MakeSliceHlo()
130 auto slice = computation->AddInstruction(HloInstruction::CreateSlice( in MakeDynamicSliceHlo()
158 auto slice = computation->AddInstruction(HloInstruction::CreateSlice( in MakeDynamicUpdateSliceHlo()
Ddot_decomposer.cc107 HloInstruction::CreateSlice(lhs_slice_shape_r3, lhs_r3, {i, 0, 0}, in DecomposeBatchDot()
114 HloInstruction::CreateSlice(rhs_slice_shape_r3, rhs_r3, {i, 0, 0}, in DecomposeBatchDot()
Dconvolution_group_converter.cc503 auto filter_slice = add(HloInstruction::CreateSlice( in HandleConvolution()
513 auto activation_slice = add(HloInstruction::CreateSlice( in HandleConvolution()
Dalgebraic_simplifier_test.cc1685 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1767 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1818 HloInstruction* slice0 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1824 HloInstruction* slice1 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1829 HloInstruction* slice2 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1834 HloInstruction* slice3 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1839 HloInstruction* slice4 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1844 HloInstruction* slice5 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1849 HloInstruction* slice6 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
1855 HloInstruction* slice7 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
[all …]
Dbuffer_assignment_test.cc1004 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1036 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1073 HloInstruction::CreateSlice(f32vec10_, tuple_element, {0}, {10}, {1})); in TEST_F()
1108 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1140 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1177 HloInstruction::CreateSlice(f32vec10_, negate, {0}, {10}, {1})); in TEST_F()
1707 auto root = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dhlo_constant_folding_test.cc162 builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dalgebraic_simplifier.cc719 computation_->AddInstruction(HloInstruction::CreateSlice( in HandleConcatenate()
1469 computation_->AddInstruction(HloInstruction::CreateSlice( in OptimizeDotOfConcatHelper()
2755 slice, HloInstruction::CreateSlice( in TrySimplifyScalarSlice()
2808 computation_->AddInstruction(HloInstruction::CreateSlice( in TryToReorderSliceAndReshape()
2836 slice, HloInstruction::CreateSlice( in HandleSlice()
Ddynamic_dimension_inference_test.cc662 auto* slice = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dbuffer_liveness_test.cc635 HloInstruction::CreateSlice(update_shape, gte1, {0}, {3}, {1})); in BuildModule()
Dhlo_instruction.h576 static std::unique_ptr<HloInstruction> CreateSlice(
Dtuple_points_to_analysis_test.cc650 HloInstruction::CreateSlice(update_shape, input, {0}, {3}, {1})); in Run()
Dheap_simulator_test.cc141 cond_builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F()
Dhlo_instruction.cc296 instruction = CreateSlice(shape, operands(0), slice_starts, slice_limits, in CreateFromProto()
984 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateSlice( in CreateSlice() function in xla::HloInstruction
Dhlo_evaluator_test.cc1783 b.AddInstruction(HloInstruction::CreateSlice(shape, operand, in TEST_P()
Dhlo_parser.cc1258 instruction = builder->AddInstruction(HloInstruction::CreateSlice( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc110 auto* sliced_result = add(HloInstruction::CreateSlice( in PadConv()
Dcudnn_conv_padding_legalization.cc379 HloInstruction::CreateSlice(backward_conv_shape, new_backward_conv, in CanonicalizeBackwardInputConvolution()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc229 auto slice12 = builder.AddInstruction(HloInstruction::CreateSlice( in XLA_TEST_F()
523 auto slice1 = builder.AddInstruction(HloInstruction::CreateSlice( in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc468 HloInstruction::CreateSlice(slice_shape, param0, {0}, {4}, {2})); in TEST_F()