Searched defs:total_example_count (Results 1 – 5 of 5) sorted by relevance
58 OpStatsLogger* opstats_logger, std::atomic<int>* total_example_count, in TrainingDatasetProvider()115 std::atomic<int>* total_example_count, in DatasetIterator()186 std::atomic<int>* total_example_count, in AddDatasetTokenToInputs()212 std::atomic<int>* total_example_count, in AddDatasetTokenToInputsForTfLite()
38 absl::StatusOr<OutputTensors> output, std::atomic<int>* total_example_count, in CreatePlanResultFromOutput()92 std::atomic<int> total_example_count = 0; in RunPlan() local
83 std::atomic<int> total_example_count = 0; in RunPlan() local121 std::atomic<int>* total_example_count, in RunPlanInternal()
212 std::atomic<int> total_example_count = 0; in RunPlan() local
1647 const size_t total_example_count = 35; in TEST() local