Searched refs:AddTensorListSizesToTerminator (Results 1 – 1 of 1) sorted by relevance
132 AddTensorListSizesToTerminator( in AddTensorListSizesToTerminator() function158 AddTensorListSizesToTerminator<ReturnOp>(func.front(), buffer_to_size); in ModifyFunctionReturn()309 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleWhileRegionOp()362 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleIfRegionOp()364 AddTensorListSizesToTerminator<TF::YieldOp>(else_branch.front(), in HandleIfRegionOp()405 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleCaseRegionOp()408 AddTensorListSizesToTerminator<TF::YieldOp>(branch->front(), in HandleCaseRegionOp()