Searched refs:ResolveForReplica (Results 1 – 3 of 3) sorted by relevance
73 StatusOr<const ShapedBuffer*> ResolveForReplica(const GlobalDataHandle& data,
744 allocation_tracker_.ResolveForReplica(outputs[i], 0)); in ExecuteGraphParallel()913 allocation_tracker_.ResolveForReplica(result->output(), 0)); in Execute()942 allocation_tracker_.ResolveForReplica(arg->data(), 0)); in TransferToClient()1121 allocation_tracker_.ResolveForReplica(arg->data(), 0)); in GetShape()
184 StatusOr<const ShapedBuffer*> AllocationTracker::ResolveForReplica( in ResolveForReplica() function in xla::AllocationTracker