/external/tensorflow/tensorflow/python/tpu/ |
D | tpu_sharding.py | 192 def get_sharded_shape(self, shape, shard_index=None): argument
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | split_utils.cc | 61 int64 num_shards, int64 shard_index, in ShardingSplitProvider()
|
D | dataset_test_base.cc | 646 const DatasetParams& params, int64 num_shards, int64 shard_index, in CheckSplitProviderShardedIteration()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | io_ops.cc | 111 int64 shard_index = -1; in WriteData() local 144 int64* shard_index) { in GetShardIndex()
|
D | snapshot_dataset_op.cc | 683 int64* shard_index) { in GetShardIndex() 748 int64 shard_index = 0; in GetNextInternal() local
|
/external/tensorflow/tensorflow/python/distribute/ |
D | parameter_server_strategy_v2.py | 622 def init_shard_fn(shard_index): argument
|
/external/rust/crates/crossbeam-utils/src/sync/ |
D | sharded_lock.rs | 228 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/ |
D | pyauto_utils.py | 170 def Shard(ilist, shard_index, num_shards): argument
|
/external/angle/src/tests/restricted_traces/ |
D | restricted_trace_gold_tests.py | 412 def _shard_tests(tests, shard_count, shard_index): argument
|
/external/googletest/googletest/test/ |
D | googletest-filter-unittest.py | 220 def RunWithSharding(total_shards, shard_index, command): argument
|
D | gtest_unittest.cc | 1961 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/ |
D | gtest.cc | 2702 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/ |
D | gtest.cc | 3071 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/ |
D | gtest.cc | 3147 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/ |
D | gtest.cc | 3140 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/ |
D | gtest.cc | 3156 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/ |
D | gtest.cc | 3403 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/ |
D | gtest.cc | 3228 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/ |
D | gmock-gtest-all.cc | 4099 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/ |
D | gmock-gtest-all.cc | 4519 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/ |
D | gtest_unittest.cc | 1952 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
|