Home
last modified time | relevance | path

Searched defs:shard_index (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtpu_sharding.py192 def get_sharded_shape(self, shape, shard_index=None): argument
/external/tensorflow/tensorflow/core/kernels/data/
Dsplit_utils.cc61 int64 num_shards, int64 shard_index, in ShardingSplitProvider()
Ddataset_test_base.cc646 const DatasetParams& params, int64 num_shards, int64 shard_index, in CheckSplitProviderShardedIteration()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dio_ops.cc111 int64 shard_index = -1; in WriteData() local
144 int64* shard_index) { in GetShardIndex()
Dsnapshot_dataset_op.cc683 int64* shard_index) { in GetShardIndex()
748 int64 shard_index = 0; in GetNextInternal() local
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_v2.py622 def init_shard_fn(shard_index): argument
/external/rust/crates/crossbeam-utils/src/sync/
Dsharded_lock.rs228 let shard_index = current_index & (self.shards.len() - 1); in try_read() localVariable
288 let shard_index = current_index & (self.shards.len() - 1); in read() localVariable
/external/autotest/server/cros/ap_configurators/
Dpyauto_utils.py170 def Shard(ilist, shard_index, num_shards): argument
/external/angle/src/tests/restricted_traces/
Drestricted_trace_gold_tests.py412 def _shard_tests(tests, shard_count, shard_index): argument
/external/googletest/googletest/test/
Dgoogletest-filter-unittest.py220 def RunWithSharding(total_shards, shard_index, command): argument
Dgtest_unittest.cc1961 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1975 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
/external/llvm/utils/unittest/googletest/src/
Dgtest.cc2702 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4281 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4336 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4350 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/llvm-project/llvm/utils/unittest/googletest/src/
Dgtest.cc3071 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
4723 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
4778 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
4792 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/rust/crates/quiche/deps/boringssl/src/third_party/googletest/src/
Dgtest.cc3147 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5364 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5419 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5433 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/libvpx/libvpx/third_party/googletest/src/src/
Dgtest.cc3140 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5265 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5320 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5334 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/mesa3d/src/gtest/src/
Dgtest.cc3156 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5426 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5481 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5495 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/googletest/googletest/src/
Dgtest.cc3403 const int32_t shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5958 const int32_t shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
6013 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
6027 const int32_t shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/libaom/libaom/third_party/googletest/src/googletest/src/
Dgtest.cc3228 const int32_t shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5467 const int32_t shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5522 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5537 const int32_t shard_index = shard_tests == HONOR_SHARDING_PROTOCOL in FilterTests() local
/external/fmtlib/test/
Dgmock-gtest-all.cc4099 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
5722 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
5777 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
5792 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL in FilterTests() local
/external/rust/crates/grpcio-sys/grpc/third_party/cares/cares/test/gmock-1.8.0/
Dgmock-gtest-all.cc4519 const Int32 shard_index = Int32FromEnvOrDie(kTestShardIndex, -1); in OnTestIterationStart() local
6171 const Int32 shard_index = Int32FromEnvOrDie(shard_index_env, -1); in ShouldShard() local
6226 bool ShouldRunTestOnShard(int total_shards, int shard_index, int test_id) { in ShouldRunTestOnShard()
6240 const Int32 shard_index = shard_tests == HONOR_SHARDING_PROTOCOL ? in FilterTests() local
/external/rust/crates/quiche/deps/boringssl/src/third_party/googletest/test/
Dgtest_unittest.cc1952 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local
1966 for (int shard_index = 0; shard_index < num_shards; shard_index++) { in TEST() local