Searched defs:kNumReplicas (Results 1 – 4 of 4) sorted by relevance
100 const int kNumReplicas = 2; in TestTwoReplicasOneOperand() local418 static constexpr int kNumReplicas = 2; in XLA_TEST_F() local499 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local531 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local564 static constexpr int kNumReplicas = 4; in XLA_TEST_F() local591 const int64_t kNumReplicas = 4; in XLA_TEST_F() local620 const int64_t kNumReplicas = 4; in XLA_TEST_F() local654 const int64_t kNumReplicas = 4; in XLA_TEST_F() local687 const int64_t kNumReplicas = 4; in XLA_TEST_F() local721 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