Searched refs:tensor_watches (Results 1 – 1 of 1) sorted by relevance
61 std::unordered_map<string, std::vector<string>> tensor_watches; in InsertNodes() local86 tensor_watches[tensor_name] = debug_ops; in InsertNodes()97 if (tensor_watches.empty()) { in InsertNodes()123 if (tensor_watches.find(tensor_name) == tensor_watches.end()) { in InsertNodes()152 src_output_slot, src_dt, tensor_name, tensor_watches[tensor_name], in InsertNodes()166 for (size_t i = 0; i < tensor_watches[tensor_name].size(); ++i) { in InsertNodes()167 const string& debug_op_name = tensor_watches[tensor_name][i]; in InsertNodes()