Commit 8b3c5e62 authored by Akshay Agrawal's avatar Akshay Agrawal Committed by TensorFlower Gardener
Browse files

`PartitionedCallOp`: An op for executing multi-device functions.

A `PartitionedCallOp` allows for execution of functions across multiple devices
but within a single process. It proceeds by placing and partitioning the graph
underlying a given function body, instantiating for each partitioned subgraph a
function. The yielded function shards, which together are equivalent to the
original function, are then executed.

`PartitionedCallOp` is not part of the public TensorFlow API.

PiperOrigin-RevId: 194173114
parent e871ea87
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment