Home
last modified time | relevance | path

Searched refs:CreateInfeed (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_simplifier_test.cc155 false_computation->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
Dhlo_dce_test.cc239 HloInstruction::CreateInfeed(shape, token, "")); in TEST_F()
Dwhile_loop_simplifier_test.cc223 while_body->AddInstruction(HloInstruction::CreateInfeed( in TEST_F()
Dhlo_instruction.h536 static std::unique_ptr<HloInstruction> CreateInfeed(
Dlayout_assignment_test.cc793 HloInstruction::CreateInfeed(xshape, token, "")); in TEST_F()
Dhlo_instruction.cc365 CreateInfeed(data_shape, operands(0), proto.infeed_config()); in CreateFromProto()
862 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateInfeed( in CreateInfeed() function in xla::HloInstruction
Dbuffer_assignment_test.cc2163 builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, token, "")); in TEST_F()
Dhlo_parser.cc1403 instruction = builder->AddInstruction(HloInstruction::CreateInfeed( in ParseInstructionRhs()