Searched refs:split_outputs (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_lower_io_arrays_to_elements.c | 358 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements_no_indirects() local
363 indirects, split_outputs, true); in nir_lower_io_arrays_to_elements_no_indirects()
379 hash_table_foreach(split_outputs, entry) { in nir_lower_io_arrays_to_elements_no_indirects()
387 _mesa_hash_table_destroy(split_outputs, NULL); in nir_lower_io_arrays_to_elements_no_indirects()
397 struct hash_table *split_outputs = _mesa_pointer_hash_table_create(NULL); in nir_lower_io_arrays_to_elements() local
405 indirects, split_outputs, false); in nir_lower_io_arrays_to_elements()
419 hash_table_foreach(split_outputs, entry) { in nir_lower_io_arrays_to_elements()
427 _mesa_hash_table_destroy(split_outputs, NULL); in nir_lower_io_arrays_to_elements()
|
| H A D | nir_lower_io_to_scalar.c | 385 struct hash_table *split_outputs) in lower_load_to_scalar_early() argument
395 chan_vars = get_channel_variables(split_outputs, var); in lower_load_to_scalar_early()
438 struct hash_table *split_outputs) in lower_store_output_to_scalar_early() argument
444 nir_variable **chan_vars = get_channel_variables(split_outputs, var); in lower_store_output_to_scalar_early()
481 struct hash_table *split_inputs, *split_outputs; member
550 state->split_outputs); in nir_lower_io_to_scalar_early_instr()
557 lower_store_output_to_scalar_early(b, intr, var, state->split_outputs); in nir_lower_io_to_scalar_early_instr()
577 .split_outputs = _mesa_pointer_hash_table_create(NULL), in nir_lower_io_to_scalar_early()
595 hash_table_foreach(state.split_outputs, entry) { in nir_lower_io_to_scalar_early()
603 _mesa_hash_table_destroy(state.split_outputs, NULL); in nir_lower_io_to_scalar_early()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| H A D | ops_defs.py | 106 split_outputs = tf.raw_ops.SplitV(
111 left_padding = split_outputs[0]
112 core = split_outputs[1]
113 right_padding = split_outputs[2]
|
| /aosp_15_r20/external/federated-compute/fcp/artifact_building/ |
| H A D | graph_helpers.py | 176 split_outputs: bool = False,
214 type_checks.check_type(split_outputs, bool, name='split_outputs')
270 if split_outputs:
324 if split_outputs:
|
| H A D | federated_compute_plan_builder.py | 447 split_outputs=True,
640 split_outputs=True,
1053 split_outputs=True,
1282 split_outputs=False,
1775 split_outputs=True,
|
| H A D | graph_helpers_test.py | 170 'work', work, ([ds], [v]), split_outputs=True
|