Searched refs:CreateRecv (Results 1 – 8 of 8) sorted by relevance
199 builder.AddInstruction(HloInstruction::CreateRecv(shape, token1, 1)); in TEST_F()
156 auto* recv = true_computation->AddInstruction(HloInstruction::CreateRecv( in TEST_F()
206 HloInstruction::CreateRecv(ShapeUtil::MakeShape(F32, {1}), token, in TEST_F()
869 static std::unique_ptr<HloInstruction> CreateRecv(
282 instruction = CreateRecv(shape.tuple_shapes(0), operands(0), in CreateFromProto()1412 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateRecv( in CreateRecv() function in xla::HloInstruction
1241 HloInstruction::CreateRecv(scalar_shape_, token, /*channel_id=*/0)); in TEST_P()
1849 return builder->AddInstruction(HloInstruction::CreateRecv( in CreateInstruction()
1932 [`HloInstruction::CreateRecv` and `HloInstruction::CreateRecvDone`](https://www.tensorflow.org/code…