Searched refs:all_arguments (Results 1 – 1 of 1) sorted by relevance
570 std::vector<std::vector<std::vector<const ShapedBuffer*>>> all_arguments; in ExecuteGraphParallel() local621 all_arguments.push_back(replicated_arguments); in ExecuteGraphParallel()622 all_arguments.insert(all_arguments.end(), executors.size() - 1, {{}}); in ExecuteGraphParallel()655 TF_RETURN_IF_ERROR(RecordArguments(all_arguments[i].front(), stream.get(), in ExecuteGraphParallel()678 executable_ptrs[0], all_arguments[0], execute_backend_.get(), in ExecuteGraphParallel()687 ExecuteParallelAndRegisterResult(executable_ptrs, all_arguments, in ExecuteGraphParallel()