Searched refs:parallel_inputs (Results 1 – 3 of 3) sorted by relevance
143 std::vector<ParallelTensor*> parallel_inputs; in ExecuteWithSpecialOps() local145 parallel_inputs.reserve(inputs.size()); in ExecuteWithSpecialOps()166 parallel_inputs.push_back(parallel_tensor.get()); in ExecuteWithSpecialOps()181 parallel_inputs.push_back(absl::get<ParallelTensor*>(input)); in ExecuteWithSpecialOps()186 parallel_device.Execute(context, parallel_inputs, operation_name, in ExecuteWithSpecialOps()
353 const std::vector<parallel_device::ParallelTensor*>& parallel_inputs,1458 const std::vector<parallel_device::ParallelTensor*>& parallel_inputs, in ExecuteFunctionAndWait() argument1464 /*inputs=*/parallel_inputs, in ExecuteFunctionAndWait()1601 StatusOr<std::vector<parallel_device::ParallelTensor*>> parallel_inputs = in ExecuteRegularOperation() local1603 if (!parallel_inputs.ok()) { in ExecuteRegularOperation()1605 parallel_inputs.status().error_message().c_str()); in ExecuteRegularOperation()1613 /*parallel_inputs=*/*parallel_inputs, /*step_id=*/step_id, in ExecuteRegularOperation()1661 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in ExecuteRegularOperation() local1662 parallel_inputs.reserve(inputs.size() + 1); in ExecuteRegularOperation()1676 parallel_inputs.push_back( in ExecuteRegularOperation()[all …]
1010 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in PrepareEmbeddingInputs() local1014 parallel_inputs.push_back(inputs[input_index]->tensor()); in PrepareEmbeddingInputs()1017 return parallel_inputs; in PrepareEmbeddingInputs()