| /external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
| D | io_memory.cpp | 44 std::vector<Tensor> Memory::get_input_tensors(int shard_index) { in get_input_tensors()
53 std::vector<Tensor> Memory::get_output_tensors(int shard_index) { in get_output_tensors()
144 for (int shard_index = 1; shard_index < modules_.size(); ++shard_index) { in prepare_io() local
154 for (int offset = 0, shard_index = 0; shard_index < modules_.size(); in prepare_io() local
178 for (int shard_index = 0; shard_index < modules_.size() - 1; ++shard_index) { in prepare_io() local
306 for (int shard_index = 1; shard_index < modules_.size(); ++shard_index) { in prepare_io() local
316 for (int offset = 0, shard_index = 0, v_stride = 1023 * 128; in prepare_io() local
348 for (int offset = 0, shard_index = 0, v_stride = 1023 * 128; in prepare_io() local
378 for (int shard_index = 0; shard_index < modules_.size() - 1; ++shard_index) { in prepare_io() local
|
| /external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/ |
| D | io_memory.cpp | 37 std::vector<Tensor> Memory::get_input_tensors(int shard_index) { in get_input_tensors()
46 std::vector<Tensor> Memory::get_output_tensors(int shard_index) { in get_output_tensors()
109 shard_index = 0, in prepare_io() local
153 shard_index = 0, in prepare_io() local
|
| /external/python/absl-py/absl/testing/tests/ |
| D | absltest_sharding_test.py | 48 shard_index, argument
178 self, total_shards, shard_index, override_expected_exit_code argument
|
| /external/rust/beto-rust/common/handle_map/src/ |
| D | lib.rs | 190 let shard_index = new_handle.get_shard_index(self.dimensions.num_shards); in try_allocate() localVariable
225 let shard_index = handle.get_shard_index(self.dimensions.num_shards); in get() localVariable
239 let shard_index = handle.get_shard_index(self.dimensions.num_shards); in get_mut() localVariable
251 let shard_index = handle.get_shard_index(self.dimensions.num_shards); in deallocate() localVariable
|
| /external/cronet/tot/testing/trigger_scripts/ |
| D | base_test_triggerer.py | 79 shard_index, argument
128 def append_additional_args(self, args, shard_index): argument
233 shard_index, argument
|
| D | perf_device_trigger.py | 125 def append_additional_args(self, args, shard_index): argument
310 def _query_swarming_for_last_shard_id(self, shard_index): argument
|
| /external/cronet/stable/testing/trigger_scripts/ |
| D | base_test_triggerer.py | 79 shard_index, argument
128 def append_additional_args(self, args, shard_index): argument
233 shard_index, argument
|
| D | perf_device_trigger.py | 125 def append_additional_args(self, args, shard_index): argument
310 def _query_swarming_for_last_shard_id(self, shard_index): argument
|
| /external/cronet/stable/testing/scripts/ |
| D | run_isolated_script_test.py | 68 def generate_sharding_args(self, total_shards, shard_index): argument
95 def generate_sharding_args(self, total_shards, shard_index): argument
|
| D | gpu_integration_test_adapter.py | 21 def generate_sharding_args(self, total_shards, shard_index): argument
|
| D | run_telemetry_as_googletest.py | 37 def generate_sharding_args(self, total_shards, shard_index): argument
|
| /external/cronet/tot/testing/scripts/ |
| D | run_isolated_script_test.py | 68 def generate_sharding_args(self, total_shards, shard_index): argument
95 def generate_sharding_args(self, total_shards, shard_index): argument
|
| D | gpu_integration_test_adapter.py | 21 def generate_sharding_args(self, total_shards, shard_index): argument
|
| D | run_telemetry_as_googletest.py | 37 def generate_sharding_args(self, total_shards, shard_index): argument
|
| /external/cronet/stable/build/android/pylib/base/ |
| D | test_exception.py | 11 def __init__(self, shard_index, total_shards): argument
|
| /external/angle/build/android/pylib/base/ |
| D | test_exception.py | 11 def __init__(self, shard_index, total_shards): argument
|
| /external/cronet/tot/build/android/pylib/base/ |
| D | test_exception.py | 11 def __init__(self, shard_index, total_shards): argument
|
| /external/webrtc/test/ |
| D | test_main_lib.cc | 140 std::string shard_index = in Init() local
|
| /external/angle/src/tests/ |
| D | run_perf_tests.py | 68 def _shard_tests(tests, shard_count, shard_index): argument
453 def _split_shard_samples(tests, samples_per_test, shard_count, shard_index): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.cc | 147 int64_t shard_index = -1; in WriteData() local
180 int64_t* shard_index) { in GetShardIndex()
380 int64_t shard_index = 0; in GetNextInternal() local
451 bool use_shard_func, int64_t* shard_index) in GetShardIndex()
|
| /external/cronet/stable/build/android/pylib/local/machine/ |
| D | local_machine_hostside_test_run.py | 86 def _ApplyExternalSharding(tests, shard_index, total_shards): argument
|
| /external/angle/build/android/pylib/local/machine/ |
| D | local_machine_hostside_test_run.py | 86 def _ApplyExternalSharding(tests, shard_index, total_shards): argument
|
| /external/cronet/tot/build/android/pylib/local/machine/ |
| D | local_machine_hostside_test_run.py | 86 def _ApplyExternalSharding(tests, shard_index, total_shards): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_sharding.py | 184 def get_sharded_shape(self, shape, shard_index=None): argument
|
| /external/rust/android-crates-io/crates/crossbeam-utils/src/sync/ |
| D | sharded_lock.rs | 230 let shard_index = current_index & (self.shards.len() - 1); in try_read() localVariable
290 let shard_index = current_index & (self.shards.len() - 1); in read() localVariable
|