Searched refs:parallel_inputs (Results 1 – 1 of 1) sorted by relevance
141 std::vector<ParallelTensor*> parallel_inputs; in ExecuteWithSpecialOps() local143 parallel_inputs.reserve(inputs.size()); in ExecuteWithSpecialOps()158 parallel_inputs.push_back(parallel_tensor.get()); in ExecuteWithSpecialOps()161 parallel_inputs.push_back(absl::get<ParallelTensor*>(input)); in ExecuteWithSpecialOps()166 parallel_device.Execute(context, parallel_inputs, operation_name, in ExecuteWithSpecialOps()