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