Searched refs:after_all (Results 1 – 11 of 11) sorted by relevance
5 %0 = "mhlo.after_all"(%arg0) : (!mhlo.token) -> !mhlo.token loc(unknown)16 %0 = "mhlo.after_all"(%arg0) : (!mhlo.token) -> !mhlo.token loc("AfterAll")27 %0 = "mhlo.after_all"(%arg0) : (!mhlo.token) -> !mhlo.token loc("name@function")38 %0 = "mhlo.after_all"(%arg0) : (!mhlo.token) -> !mhlo.token loc("file_name":2:8)
6 %0 = "mhlo.after_all"(%arg0, %arg1) : (!mhlo.token, !mhlo.token) -> !mhlo.token
31 // CHECK-NEXT: "mhlo.after_all"([[VAL_0]], [[VAL_1]]) : (!mhlo.token, !mhlo.token) -> !mhlo.token
36 …// CHECK: [[SEND_SINK_TOKEN:%.*]] = "mhlo.after_all"([[SEND_ARG0_TOKEN]], [[SEND_ARG1_TOKEN]])72 …// CHECK: [[RECV_SINK_TOKEN:%.*]] = "mhlo.after_all"([[RECV_RETVAL0_TOKEN]], [[RECV_RETVAL1_T…117 // CHECK-NOT: "mhlo.after_all"134 // CHECK-NOT: "mhlo.after_all"145 // create any `mhlo.after_all` ops.153 // CHECK-NOT: "mhlo.after_all"155 // CHECK-NOT: "mhlo.after_all"
249 Status HandleAfterAll(HloInstruction* after_all) override;
1839 Status HloEvaluator::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument1840 evaluated_[after_all] = LiteralUtil::CreateToken(); in HandleAfterAll()
168 Status HandleAfterAll(HloInstruction* after_all) override;
2740 Status IrEmitter::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument2741 TF_RET_CHECK(ByteSizeOf(after_all->shape()) == 0); in HandleAfterAll()2743 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(after_all)); in HandleAfterAll()
214 Status HandleAfterAll(HloInstruction* after_all) override;
3228 Status IrEmitterUnnested::HandleAfterAll(HloInstruction* after_all) { in HandleAfterAll() argument
508 def HLO_AfterAllOp : HLO_Op<"after_all", [NoSideEffect]> {