(data_list, thread_num)
| 23 | |
| 24 | |
| 25 | def _split_input_data(data_list, thread_num): |
| 26 | size = tf.shape(data_list)[0] |
| 27 | split_size = [size // thread_num] * (thread_num - 1) |
| 28 | if thread_num == 1: |
| 29 | split_size += [size] |
| 30 | else: |
| 31 | split_size += [-1] |
| 32 | split_data_list = tf.split(data_list, split_size) |
| 33 | return split_data_list |
| 34 | |
| 35 | |
| 36 | def _get_sparse_feature(nodes_or_edges, feature_names, op, thread_num, |
no outgoing calls
no test coverage detected