Searched refs:all_arguments (Results 1 – 1 of 1) sorted by relevance
618 std::vector<std::vector<std::vector<const ShapedBuffer*>>> all_arguments; in ExecuteGraphParallel() local669 all_arguments.push_back(replicated_arguments); in ExecuteGraphParallel()670 all_arguments.insert(all_arguments.end(), executors.size() - 1, {{}}); in ExecuteGraphParallel()700 TF_RETURN_IF_ERROR(RecordArguments(all_arguments[i].front(), stream.get(), in ExecuteGraphParallel()722 ExecuteAndRegisterResult(executable_ptrs[0], all_arguments[0], in ExecuteGraphParallel()729 executable_ptrs, all_arguments, execute_backend_.get(), in ExecuteGraphParallel()