Searched refs:wrapped_results (Results 1 – 2 of 2) sorted by relevance
741 std::vector<std::vector<std::unique_ptr<PyTpuBuffer>>> wrapped_results( in ExecuteOnLocalDevices() local745 TF_ASSIGN_OR_RETURN(wrapped_results[i], in ExecuteOnLocalDevices()748 wrapped_results[i].push_back(std::move(results[i].buffer)); in ExecuteOnLocalDevices()751 return wrapped_results; in ExecuteOnLocalDevices()
2033 std::vector<std::vector<std::unique_ptr<PjRtBuffer>>> wrapped_results( in Execute() local2051 wrapped_results[i] = std::move(statusor.ValueOrDie()); in Execute()2053 return wrapped_results; in Execute()