Searched refs:IsAsyncComputation (Results 1 – 9 of 9) sorted by relevance
571 CHECK(!IsCustomCallComputation() && !IsAsyncComputation()); in SetFusionInstruction()585 CHECK(!IsFusionComputation() && !IsAsyncComputation()); in SetCustomCallInstruction()591 bool IsAsyncComputation() const { return !async_instructions_.empty(); } in IsAsyncComputation() function
106 if (computation->IsAsyncComputation()) { in FlattenNode()
156 !instruction->parent()->IsAsyncComputation()) { in Run()
201 CHECK(async_context->IsAsyncComputation()); in CalculateBufferStartEndMap()
366 !node.computation()->IsAsyncComputation() && in IsFlattened()
284 if (options.syntax_sugar_async_ops() && computation->IsAsyncComputation()) { in ToCord()
112 if (IsAsyncComputation()) { in ~HloComputation()
290 if (computation->IsAsyncComputation()) { in ClearAsyncComputationInstruction()
362 CHECK(node.computation()->IsAsyncComputation() || callsites.size() == 1) in CalculateComputationNestLevel()