Home
last modified time | relevance | path

Searched refs:CreatePad (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcublas_gemm_pad_for_tensor_cores.cc89 HloInstruction::CreatePad(new_lshape, lhs, zero_float, l_padding_config)); in PadForTensorCores()
93 HloInstruction::CreatePad(new_rshape, rhs, zero_float, r_padding_config)); in PadForTensorCores()
Dtree_reduction_rewriter.cc134 return hlo->parent()->AddInstruction(HloInstruction::CreatePad( in RewriteReduction()
Dcudnn_pad_for_convolutions.cc70 HloInstruction::CreatePad(new_shape, instr, zero, pad_config)); in PadInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier_test.cc360 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
384 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
Dalgebraic_simplifier_test.cc1847 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3225 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3254 auto pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3290 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3338 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3384 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3866 auto* lhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P()
3976 auto* rhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P()
4355 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
4437 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
[all …]
Dhlo_creation_utils.cc77 return computation->AddInstruction(HloInstruction::CreatePad( in MakePadHlo()
Dhlo_evaluator_test.cc661 b.AddInstruction(HloInstruction::CreatePad( in TEST_F()
688 b.AddInstruction(HloInstruction::CreatePad( in TEST_P()
730 b.AddInstruction(HloInstruction::CreatePad(shape, input_instruction, in TEST_P()
775 b.AddInstruction(HloInstruction::CreatePad(shape, input_instruction, in TEST_P()
Dhlo_instruction.h890 static std::unique_ptr<HloInstruction> CreatePad(
Dmemory_space_assignment_test.cc2482 HloInstruction::CreatePad(shape2, mul2, padding_value, padding_config)); in TEST_P()
2558 HloInstruction::CreatePad(shape2, add1, padding_value, padding_config)); in TEST_P()
2562 HloInstruction::CreatePad(shape3, add2, padding_value, padding_config)); in TEST_P()
2568 HloInstruction::CreatePad(shape4, add3, padding_value, padding_config)); in TEST_P()
Dalgebraic_simplifier.cc1346 computation_->AddInstruction(HloInstruction::CreatePad( in HandleConcatenate()
3344 HloInstruction::CreatePad(pad->shape(), pad->mutable_operand(0), in HandlePad()
4487 computation_->AddInstruction(HloInstruction::CreatePad( in HandleDynamicUpdateSlice()
Dhlo_instruction.cc623 CreatePad(shape, operands(0), operands(1), proto.padding_config()); in CreateFromProto()
1439 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreatePad( in CreatePad() function in xla::HloInstruction
Dhlo_parser.cc2067 instruction = builder->AddInstruction(HloInstruction::CreatePad( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc273 HloInstruction::CreatePad(padded_shape, hlo, zero, padding_config)); in PadToShape()
588 concat = b->AddInstruction(HloInstruction::CreatePad( in PadFromPartialReplicateShape()
623 HloInstruction::CreatePad(padded_shape, result, zero, pad_config)); in PadFromPartialReplicateShape()
982 concat = b->AddInstruction(HloInstruction::CreatePad( in ExchangeHaloAndGetValidData()
Dconvolution_handler.cc456 HloInstruction::CreatePad(pad_shape, lhs.hlo(), zero, pad_config)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
Ddot_handler.cc811 body_b.AddInstruction(HloInstruction::CreatePad( in PartitionBaseCase()
822 body_b.AddInstruction(HloInstruction::CreatePad( in PartitionBaseCase()
Dspmd_partitioner.cc757 : state_.b->AddInstruction(HloInstruction::CreatePad( in ReshardAsWindowedInput()
2669 sharded_pad = b_.AddInstruction(hlo->CreatePad(sharded_pad_shape, in HandlePad()