Home
last modified time | relevance | path

Searched defs:all_labels (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dobject_detection_stage.h58 void SetAllLabels(const std::vector<std::string>& all_labels) { in SetAllLabels()
Dtopk_accuracy_eval_stage.h48 void SetTaskInfo(const std::vector<std::string>& all_labels, in SetTaskInfo()
Dimage_classification_stage.h51 void SetAllLabels(const std::vector<std::string>& all_labels) { in SetAllLabels()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Deinsum_spmd_expander.cc399 absl::flat_hash_set<char> all_labels; in MaybeRelayoutInputs() local