Searched defs:parallel_inputs (Results 1 – 3 of 3) sorted by relevance
1458 const std::vector<parallel_device::ParallelTensor*>& parallel_inputs, in ExecuteFunctionAndWait()1601 StatusOr<std::vector<parallel_device::ParallelTensor*>> parallel_inputs = in ExecuteRegularOperation() local1661 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in ExecuteRegularOperation() local
1010 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in PrepareEmbeddingInputs() local
143 std::vector<ParallelTensor*> parallel_inputs; in ExecuteWithSpecialOps() local