Commit c44566cd authored by Ayush Dubey's avatar Ayush Dubey Committed by TensorFlower Gardener
Browse files

Set collective_executor in function CallOp from OpKernelContext. 

Without this change, CollectiveOps in functions cannot execute.

PiperOrigin-RevId: 210955255
parent 446a7b54
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment