/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cublas_gemm_pad_for_tensor_cores.cc | 89 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()
|
D | tree_reduction_rewriter.cc | 134 return hlo->parent()->AddInstruction(HloInstruction::CreatePad( in RewriteReduction()
|
D | cudnn_pad_for_convolutions.cc | 70 HloInstruction::CreatePad(new_shape, instr, zero, pad_config)); in PadInstruction()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_verifier_test.cc | 360 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F() 384 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
|
D | algebraic_simplifier_test.cc | 1847 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 …]
|
D | hlo_creation_utils.cc | 77 return computation->AddInstruction(HloInstruction::CreatePad( in MakePadHlo()
|
D | hlo_evaluator_test.cc | 661 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()
|
D | hlo_instruction.h | 890 static std::unique_ptr<HloInstruction> CreatePad(
|
D | memory_space_assignment_test.cc | 2482 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()
|
D | algebraic_simplifier.cc | 1346 computation_->AddInstruction(HloInstruction::CreatePad( in HandleConcatenate() 3344 HloInstruction::CreatePad(pad->shape(), pad->mutable_operand(0), in HandlePad() 4487 computation_->AddInstruction(HloInstruction::CreatePad( in HandleDynamicUpdateSlice()
|
D | hlo_instruction.cc | 623 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
|
D | hlo_parser.cc | 2067 instruction = builder->AddInstruction(HloInstruction::CreatePad( in ParseInstructionRhs()
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | spmd_partitioner_util.cc | 273 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()
|
D | convolution_handler.cc | 456 HloInstruction::CreatePad(pad_shape, lhs.hlo(), zero, pad_config)); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
|
D | dot_handler.cc | 811 body_b.AddInstruction(HloInstruction::CreatePad( in PartitionBaseCase() 822 body_b.AddInstruction(HloInstruction::CreatePad( in PartitionBaseCase()
|
D | spmd_partitioner.cc | 757 : state_.b->AddInstruction(HloInstruction::CreatePad( in ReshardAsWindowedInput() 2669 sharded_pad = b_.AddInstruction(hlo->CreatePad(sharded_pad_shape, in HandlePad()
|