Searched refs:CheckAsyncOpComputationShapes (Results 1 – 1 of 1) sorted by relevance
1388 Status CheckAsyncOpComputationShapes(const HloInstruction* async_op, in CheckAsyncOpComputationShapes() function1456 CheckAsyncOpComputationShapes(async_start, async_start->shape())); in HandleAsyncStart()1482 CheckAsyncOpComputationShapes(async_update, async_update->shape())); in HandleAsyncUpdate()1488 TF_RETURN_IF_ERROR(CheckAsyncOpComputationShapes( in HandleAsyncDone()