Home
last modified time | relevance | path

Searched defs:shard (Results 1 – 25 of 91) sorted by relevance

1234

/external/pytorch/torch/csrc/distributed/autograd/context/
Dcontainer.cpp114 auto& shard = getShard(context_id); in getOrCreateContext() local
147 auto& shard = getShard(context_id); in newContext() local
172 auto& shard = getShard(current_context_id_); in currentContext() local
183 auto& shard = getShard(context_id); in releaseContextIfPresent() local
202 auto& shard = getShard(context_id); in releaseContext() local
269 DistAutogradContainer::ContextsShard& shard, in eraseContextIdAndReset()
281 auto& shard = getShard(context_id); in isValidContext() local
290 auto& shard = getShard(context_id); in retrieveContext() local
321 for (const auto& shard : autograd_contexts_) { in numAutogradContexts() local
/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/lib/iomgr/
Dtimer_generic.cc235 static grpc_core::Timestamp compute_min_deadline(timer_shard* shard) { in compute_min_deadline()
259 timer_shard* shard = &g_shards[i]; in timer_list_init() local
278 timer_shard* shard = &g_shards[i]; in timer_list_shutdown() local
314 static void note_deadline_change(timer_shard* shard) { in note_deadline_change()
332 timer_shard* shard = &g_shards[grpc_core::HashPointer(timer, g_num_shards)]; in timer_init() local
439 timer_shard* shard = &g_shards[grpc_core::HashPointer(timer, g_num_shards)]; in timer_cancel() local
468 static bool refill_heap(timer_shard* shard, grpc_core::Timestamp now) { in refill_heap()
508 static grpc_timer* pop_one(timer_shard* shard, grpc_core::Timestamp now) { in pop_one()
542 static size_t pop_timers(timer_shard* shard, grpc_core::Timestamp now, in pop_timers()
Dtcp_client_posix.cc257 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in on_writable() local
392 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in grpc_tcp_client_create_from_prepared_fd() local
438 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in tcp_cancel_connect() local
/external/grpc-grpc/src/core/lib/iomgr/
Dtimer_generic.cc235 static grpc_core::Timestamp compute_min_deadline(timer_shard* shard) { in compute_min_deadline()
259 timer_shard* shard = &g_shards[i]; in timer_list_init() local
278 timer_shard* shard = &g_shards[i]; in timer_list_shutdown() local
314 static void note_deadline_change(timer_shard* shard) { in note_deadline_change()
332 timer_shard* shard = &g_shards[grpc_core::HashPointer(timer, g_num_shards)]; in timer_init() local
439 timer_shard* shard = &g_shards[grpc_core::HashPointer(timer, g_num_shards)]; in timer_cancel() local
468 static bool refill_heap(timer_shard* shard, grpc_core::Timestamp now) { in refill_heap()
508 static grpc_timer* pop_one(timer_shard* shard, grpc_core::Timestamp now) { in pop_one()
542 static size_t pop_timers(timer_shard* shard, grpc_core::Timestamp now, in pop_timers()
Dtcp_client_posix.cc260 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in on_writable() local
396 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in grpc_tcp_client_create_from_prepared_fd() local
442 struct ConnectionShard* shard = &(*g_connection_shards)[shard_number]; in tcp_cancel_connect() local
/external/libtextclassifier/native/utils/grammar/utils/
Drules.cc208 const int8 max_whitespace_gap, const bool case_sensitive, const int shard, in ExpandOptionals()
267 const std::vector<RhsElement>& rhs, int shard) { in ResolveFillers()
315 const std::vector<RhsElement>& rhs, int shard) { in OptimizeRhs()
322 const int shard) { in Add()
344 const int shard) { in Add()
375 const bool case_sensitive, const int shard) { in AddWithExclusion()
385 const bool case_sensitive, const int shard) { in AddAssertion()
394 const bool case_sensitive, const int shard) { in AddValueMapping()
402 const bool case_sensitive, const int shard) { in AddValueMapping()
/external/autotest/client/site_tests/graphics_parallel_dEQP/
Dgenerate_controlfiles.py108 def get_controlfilename(test, shard=0): argument
124 def get_name(test, shard): argument
131 def get_testname(test, shard=0): argument
/external/grpc-grpc/src/core/lib/event_engine/posix_engine/
Dtimer.cc58 Shard& shard = shards_[i]; in TimerList() local
95 void TimerList::NoteDeadlineChange(Shard* shard) { in NoteDeadlineChange()
111 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerInit() local
164 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerCancel() local
/external/boringssl/src/util/
Dall_tests.json4 "shard": true boolean
46 "shard": true boolean
50 "shard": true boolean
54 "shard": true boolean
/external/cronet/tot/third_party/boringssl/src/util/
Dall_tests.json4 "shard": true boolean
46 "shard": true boolean
50 "shard": true boolean
54 "shard": true boolean
/external/cronet/stable/third_party/boringssl/src/util/
Dall_tests.json4 "shard": true boolean
46 "shard": true boolean
50 "shard": true boolean
54 "shard": true boolean
/external/rust/android-crates-io/crates/grpcio-sys/grpc/src/core/lib/event_engine/posix_engine/
Dtimer.cc58 Shard& shard = shards_[i]; in TimerList() local
95 void TimerList::NoteDeadlineChange(Shard* shard) { in NoteDeadlineChange()
111 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerInit() local
164 Shard* shard = &shards_[grpc_core::HashPointer(timer, num_shards_)]; in TimerCancel() local
/external/compiler-rt/lib/sanitizer_common/tests/
Dsanitizer_thread_registry_test.cc163 uptr shard; // started from 1. member
170 uptr shard = (uptr)arg; in OnJoined() local
174 uptr shard = (uptr)arg; in OnStarted() local
178 uptr shard = (uptr)arg; in OnCreated() local
/external/aws-sdk-java-v2/services/kinesis/src/it/java/software/amazon/awssdk/services/kinesis/
DKinesisIntegrationTests.java121 Shard shard = shards.get(0); in testCreatePutGetDelete() local
135 private void testGets(final String streamName, final Shard shard) throws InterruptedException { in testGets()
/external/pytorch/torch/distributed/_shard/sharding_spec/
Dapi.py77 def shard( member in ShardingSpec
179 def shard( member in EnumerableShardingSpec
/external/pytorch/torch/distributed/_shard/
Dsharder.py18 def shard(self, module: nn.Module) -> nn.Module: member in Sharder
/external/bazelbuild-rules_android/src/common/golang/
Dshard_test.go15 package shard package
Dzipshard_test.go15 package shard package
Dshard.go16 package shard package
/external/rust/android-crates-io/crates/dashmap/src/
Dread_only.rs68 let shard = unsafe { self.map._get_read_shard(idx) }; in contains_key() localVariable
83 let shard = unsafe { self.map._get_read_shard(idx) }; in get() localVariable
98 let shard = unsafe { self.map._get_read_shard(idx) }; in get_key_value() localVariable
Dlib.rs99 let shard = shard.read(); in clone() localVariable
923 let mut shard = unsafe { self._yield_write_shard(idx) }; in _insert() localVariable
939 let mut shard = unsafe { self._yield_write_shard(idx) }; in _remove() localVariable
953 let mut shard = unsafe { self._yield_write_shard(idx) }; in _remove_if() localVariable
980 let mut shard = unsafe { self._yield_write_shard(idx) }; in _remove_if_mut() localVariable
1015 let shard = unsafe { self._yield_read_shard(idx) }; in _get() localVariable
1037 let shard = unsafe { self._yield_write_shard(idx) }; in _get_mut() localVariable
1059 let shard = match unsafe { self._try_yield_read_shard(idx) } { in _try_get() localVariable
1084 let shard = match unsafe { self._try_yield_write_shard(idx) } { in _try_get_mut() localVariable
1152 let shard = unsafe { self._yield_write_shard(idx) }; in _entry() localVariable
[all …]
/external/rust/android-crates-io/crates/sharded-slab/src/
Dpool.rs104 shard: &'a Shard<T, C>, field
121 shard: &'a Shard<T, C>, field
679 let shard = self.shards.get(tid.as_usize())?; in get() localVariable
770 let shard = self.shards.get(tid.as_usize())?; in get_owned() localVariable
823 let shard = self.shards.get(tid.as_usize()); in clear() localVariable
1236 fn shard(&self) -> Option<&Shard<T, C>> { in shard() function
Dlib.rs215 mod shard; module
249 shard: &'a Shard<Option<T>, C>, field
519 let shard = self.shards.get(tid.as_usize()); in remove() localVariable
585 let shard = self.shards.get(tid.as_usize())?; in take() localVariable
612 let shard = self.shards.get(tid.as_usize())?; in get() localVariable
710 let shard = self.shards.get(tid.as_usize())?; in get_owned() localVariable
/external/tensorflow/tensorflow/core/kernels/
Drecord_yielder.cc136 Shard* shard = &shards[i]; in MainLoop() local
198 void RecordYielder::ShardLoop(Shard* shard) { in ShardLoop()
/external/libtextclassifier/native/utils/grammar/parsing/
Dmatcher.cc269 for (const RulesSet_::Rules* shard : rules_shards_) { in AddTerminal() local
405 for (const RulesSet_::Rules* shard : rules_shards_) { in ProcessPendingSet() local
425 for (const RulesSet_::Rules* shard : rules_shards_) { in ProcessPendingSet() local

1234