Searched refs:AddTensorListSizesToTerminator (Results 1 – 1 of 1) sorted by relevance
116 AddTensorListSizesToTerminator( in AddTensorListSizesToTerminator() function142 AddTensorListSizesToTerminator<ReturnOp>(func.front(), buffer_to_size); in ModifyFunctionReturn()293 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleWhileRegionOp()346 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleIfRegionOp()348 AddTensorListSizesToTerminator<TF::YieldOp>(else_branch.front(), in HandleIfRegionOp()389 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleCaseRegionOp()392 AddTensorListSizesToTerminator<TF::YieldOp>(branch->front(), in HandleCaseRegionOp()