Searched refs:split_inputs (Results 1 – 4 of 4) sorted by relevance
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_io_arrays_to_elements.c | 369 struct hash_table *split_inputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements_no_indirects() local
379 indirects, split_inputs, true); in nir_lower_io_arrays_to_elements_no_indirects()
382 hash_table_foreach(split_inputs, entry) { in nir_lower_io_arrays_to_elements_no_indirects()
398 _mesa_hash_table_destroy(split_inputs, NULL); in nir_lower_io_arrays_to_elements_no_indirects()
407 struct hash_table *split_inputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements() local
419 indirects, split_inputs, false); in nir_lower_io_arrays_to_elements()
422 hash_table_foreach(split_inputs, entry) { in nir_lower_io_arrays_to_elements()
437 _mesa_hash_table_destroy(split_inputs, NULL); in nir_lower_io_arrays_to_elements()
|
| D | nir_lower_io_to_scalar.c | 179 nir_variable *var, struct hash_table *split_inputs, in lower_load_to_scalar_early() argument
190 chan_vars = get_channel_variables(split_inputs, var); in lower_load_to_scalar_early()
291 struct hash_table *split_inputs, *split_outputs; member
356 lower_load_to_scalar_early(b, intr, var, state->split_inputs, in nir_lower_io_to_scalar_early_instr()
383 .split_inputs = _mesa_pointer_hash_table_create(NULL), in nir_lower_io_to_scalar_early()
395 hash_table_foreach(state.split_inputs, entry) { in nir_lower_io_to_scalar_early()
410 _mesa_hash_table_destroy(state.split_inputs, NULL); in nir_lower_io_to_scalar_early()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 586 std::vector<Tensor> split_inputs; in Compute() local
595 TF_RETURN_IF_ERROR(Split(context, data_t, sizes, &split_inputs)); in Compute()
630 runs_it->second.context->set_output(0, split_inputs[i]); in Compute()
639 WaitingTensor{deadline_micros, split_inputs[i]}) in Compute()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1801 split_inputs = [
1806 transposed_inputs = [list(i) for i in zip(*split_inputs)]
|