Searched defs:kNumReplicas (Results 1 – 4 of 4) sorted by relevance
100 const int kNumReplicas = 2; in TestTwoReplicasOneOperand() local494 static constexpr int kNumReplicas = 2; in XLA_TEST_F() local574 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local606 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local639 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local666 const int64_t kNumReplicas = 4; in XLA_TEST_F() local695 const int64_t kNumReplicas = 4; in XLA_TEST_F() local729 const int64_t kNumReplicas = 4; in XLA_TEST_F() local762 const int64_t kNumReplicas = 4; in XLA_TEST_F() local796 const int64_t kNumReplicas = 4; in XLA_TEST_F() local[all …]
53 const int kNumReplicas = 4; in XLA_TEST_F() local
31 static constexpr const char* const kNumReplicas = "num_replicas"; variable
28 /* static */ constexpr const char* const AutoShardDatasetOp::kNumReplicas; member in tensorflow::data::experimental::AutoShardDatasetOp