Lines Matching defs:node_
320 node_.set_id(id);
321 node_.set_name(node->name());
322 node_.set_op(node->op());
323 node_.set_float_ops(0);
326 (*node_.mutable_attrs())[attr.first].MergeFrom(attr.second);
356 void AddFloatOps(int64 float_ops) { node_.set_float_ops(float_ops); }
366 const string& name() const { return node_.name(); }
367 int64 id() const { return node_.id(); }
368 const string& op() const { return node_.op(); }
369 const ProfileNode& node() { return node_; }
376 if (node_.canonical_device().empty() || node_.host_device().empty()) {
384 node_.clear_shape();
385 node_.mutable_shape()->Reserve(shape().size());
387 node_.add_shape(s);
390 node_.clear_op_types();
391 node_.mutable_op_types()->Reserve(op_types().size());
393 node_.add_op_types(t);
396 node_.clear_execs();
398 auto& exec_pb = (*node_.mutable_execs())[exec.first];
402 node_.clear_inputs();
404 (*node_.mutable_inputs())[inp.first] = nodes_map.at(inp.second)->id();
407 node_.clear_input_shapes();
409 auto& shape = (*node_.mutable_input_shapes())[s.first];
415 node_.clear_output_shapes();
417 auto& shape = (*node_.mutable_output_shapes())[s.first];
423 node_.clear_src_output_index();
426 (*node_.mutable_src_output_index())[id] = s.second;
430 node_.clear_trace();
431 node_.mutable_trace()->MergeFrom(call_stack_->code_def());
433 return node_;
438 node_.Clear();
439 node_.MergeFrom(node);
444 op_types_.insert(node_.op_types().begin(), node_.op_types().end());
447 for (int64 s : node_.shape()) {
681 return node_.float_ops();
684 return node_.float_ops() * run_count(step);
687 string canonical_device() const { return node_.canonical_device(); }
688 string host_device() const { return node_.host_device(); }
692 const auto it = node_.attrs().find(name);
693 if (it == node_.attrs().end()) {
738 ProfileNode node_;
