Home
last modified time | relevance | path

Searched refs:CreateRecv (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconditional_simplifier_test.cc142 auto* recv = true_computation->AddInstruction(HloInstruction::CreateRecv( in TEST_F()
Dhlo_reachability_test.cc198 builder.AddInstruction(HloInstruction::CreateRecv(shape, token1, 1)); in TEST_F()
Dwhile_loop_simplifier_test.cc206 HloInstruction::CreateRecv(ShapeUtil::MakeShape(F32, {1}), token, in TEST_F()
Dbuffer_liveness_test.cc333 HloInstruction::CreateRecv(vec_, token, /*channel_id=*/0)); in TEST_F()
Dhlo_instruction.h565 static std::unique_ptr<HloInstruction> CreateRecv(
Dtuple_points_to_analysis_test.cc362 auto recv = builder.AddInstruction(HloInstruction::CreateRecv( in TEST_F()
Dhlo_instruction.cc196 instruction = CreateRecv(shape.tuple_shapes(0), operands(0), in CreateFromProto()
892 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateRecv( in CreateRecv() function in xla::HloInstruction
Dhlo_dataflow_analysis_test.cc1206 HloInstruction::CreateRecv(scalar_shape_, token, /*channel_id=*/0)); in TEST_P()
Dhlo_parser.cc969 instruction = builder->AddInstruction(HloInstruction::CreateRecv( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md1642 [`HloInstruction::CreateRecv` and `HloInstruction::CreateRecvDone`](https://www.tensorflow.org/code…