Searched refs:AddTensorListSizesToTerminator (Results 1 – 1 of 1) sorted by relevance
117 AddTensorListSizesToTerminator( in AddTensorListSizesToTerminator() function143 auto output_buffer_to_size = AddTensorListSizesToTerminator<func::ReturnOp>( in ModifyFunctionReturn()296 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleWhileRegionOp()349 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleIfRegionOp()351 AddTensorListSizesToTerminator<TF::YieldOp>(else_branch.front(), in HandleIfRegionOp()392 auto output_buffer_to_size = AddTensorListSizesToTerminator<TF::YieldOp>( in HandleCaseRegionOp()395 AddTensorListSizesToTerminator<TF::YieldOp>(branch->front(), in HandleCaseRegionOp()