Searched refs:argument_handles (Results 1 – 4 of 4) sorted by relevance
228 absl::Span<LocalShapedBuffer* const> argument_handles) { in Execute() argument240 argument_buffers.reserve(argument_handles.size()); in Execute()241 for (auto& handle : argument_handles) { in Execute()265 absl::Span<const std::vector<LocalShapedBuffer*>> argument_handles) { in ExecutePerReplica() argument268 if (argument_handles.size() != num_replicas()) { in ExecutePerReplica()271 argument_handles.size(), num_devices); in ExecutePerReplica()273 if (argument_handles.size() > num_devices) { in ExecutePerReplica()276 argument_handles.size(), num_devices); in ExecutePerReplica()282 auto execute = [this, &argument_handles, &results](int replica) { in ExecutePerReplica()288 argument_buffers.reserve(argument_handles[replica].size()); in ExecutePerReplica()[all …]
206 absl::Span<XrtAllocation* const> argument_handles) { in Execute() argument224 TF_RET_CHECK(argument_handles.size() == arguments.size()); in Execute()233 inputs.insert({arguments[i], argument_handles[i]->handle()}); in Execute()
100 absl::Span<XrtAllocation* const> argument_handles);
138 absl::Span<LocalShapedBuffer* const> argument_handles);144 absl::Span<const std::vector<LocalShapedBuffer*> > argument_handles);