Searched defs:output_indexes (Results 1 – 9 of 9) sorted by relevance
29 … const std::vector<int> &output_indexes) { in ClassificationTrainAccuracyMonitor()61 auto output_indexes = accuracy_metrics_->output_indexes_; in StepEnd() local
26 const std::vector<int> &output_indexes) in AccuracyMetrics()
53 std::vector<uint32_t> output_indexes; in GetOutputNodeIdx() local
27 const std::vector<int> &output_indexes) { in AccuracyMetrics()
27 const std::vector<int> &output_indexes) { in TrainAccuracy()
309 std::vector<uint32_t> output_indexes = model_->graph_.all_nodes_.at(node_index)->output_indices_; in IsNodeSubGraphHead() local328 std::vector<uint32_t> output_indexes = node_list_.at(node_index)->output_indices_; in IsNodeSubGraphHeadWithRoot() local
87 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> output_indexes; member
379 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()411 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()1275 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateOutputPlaceholder()
453 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in GetParamAndOutputIndex()