Searched refs:array_split (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1_test.py | 300 for i, batch in enumerate(np.array_split(_TEST_DATA, 4)):
316 for i, batch in enumerate(np.array_split(_TEST_DATA, 4)):
340 batches = np.array_split(_TEST_DATA, 4)
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_split_vars.c | 363 struct array_split { struct
368 struct array_split *splits; argument
377 struct array_split root_split;
516 struct array_split *split, in create_split_array_vars()
544 split->splits = rzalloc_array(mem_ctx, struct array_split, in create_split_array_vars()
831 struct array_split *split = &info->root_split; in split_array_access_impl()
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | hdf5_format.py | 835 chunked_data = np.array_split(data_npy, num_chunks)
840 chunked_data = np.array_split(data_npy, num_chunks)
|
| /external/mesa3d/src/compiler/glsl/ |
| D | glsl_parser_extras.cpp | 2377 bool array_split = optimize_split_arrays(ir, linked); in do_common_optimization() local
2378 if (array_split) in do_common_optimization()
2380 progress |= array_split; in do_common_optimization()
|