Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier_test.cc306 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
330 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
Dalgebraic_simplifier_test.cc1483 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
2545 builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
2578 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
2624 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3106 auto* lhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P()
3214 auto* rhs_pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P()
3588 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
3670 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_F()
4238 HloInstruction* pad = builder.AddInstruction(HloInstruction::CreatePad( in TEST_P()
Dhlo_creation_utils.cc66 return computation->AddInstruction(HloInstruction::CreatePad( in MakePadHlo()
Dhlo_evaluator_test.cc584 b.AddInstruction(HloInstruction::CreatePad( in TEST_F()
611 b.AddInstruction(HloInstruction::CreatePad( in TEST_P()
653 b.AddInstruction(HloInstruction::CreatePad(shape, input_instruction, in TEST_P()
698 b.AddInstruction(HloInstruction::CreatePad(shape, input_instruction, in TEST_P()
Dhlo_instruction.h681 static std::unique_ptr<HloInstruction> CreatePad(
Dhlo_instruction.cc480 CreatePad(shape, operands(0), operands(1), proto.padding_config()); in CreateFromProto()
1167 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreatePad( in CreatePad() function in xla::HloInstruction
Dalgebraic_simplifier.cc769 computation_->AddInstruction(HloInstruction::CreatePad( in HandleConcatenate()
2216 HloInstruction::CreatePad(pad->shape(), pad->mutable_operand(0), in HandlePad()
Dhlo_parser.cc1370 instruction = builder->AddInstruction(HloInstruction::CreatePad( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_pad_for_tensor_cores.cc70 HloInstruction::CreatePad(new_shape, instr, zero, pad_config)); in PadInstruction()