Searched defs:output_indexes (Results 1 – 10 of 10) sorted by relevance
28 … const std::vector<int> &output_indexes) { in ClassificationTrainAccuracyMonitor()60 auto output_indexes = accuracy_metrics_->output_indexes_; in StepEnd() local
26 const std::vector<int> &output_indexes) in AccuracyMetrics()
27 const std::vector<int> &output_indexes) { in AccuracyMetrics()
27 const std::vector<int> &output_indexes) { in TrainAccuracy()
94 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> output_indexes; member
281 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()314 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()436 std::vector<size_t> output_indexes; in GetSingleOpRunInfo() local
318 std::vector<uint32_t> output_indexes = model_->graph_.all_nodes_.at(node_index)->output_indices_; in IsNodeSubGraphHead() local337 std::vector<uint32_t> output_indexes = node_list_.at(node_index)->output_indices_; in IsNodeSubGraphHeadWithRoot() local
58 std::vector<size_t> output_indexes; member
828 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in GetParamAndOutputIndex()
543 const auto &output_indexes = op_run_info->base_op_run_info.output_indexes; in SetOutputDeviceAddressFlag() local