Searched refs:mutable_proto (Results 1 – 8 of 8) sorted by relevance
29 mutable_proto()->set_name(name()); in ReInit()30 mutable_proto()->clear_devices(); in ReInit()32 mutable_proto()->add_devices(node->canonical_device()); in ReInit()34 mutable_proto()->set_run_count(node->run_count(step)); in ReInit()35 mutable_proto()->set_exec_micros(node->exec_micros(step)); in ReInit()36 mutable_proto()->set_accelerator_exec_micros( in ReInit()38 mutable_proto()->set_cpu_exec_micros(node->cpu_exec_micros(step)); in ReInit()40 mutable_proto()->set_requested_bytes(node->requested_bytes(step)); in ReInit()41 mutable_proto()->set_peak_bytes(node->peak_bytes(step)); in ReInit()42 mutable_proto()->set_residual_bytes(node->residual_bytes(step)); in ReInit()[all …]
48 GraphNodeProto* mutable_proto();94 MultiGraphNodeProto* mutable_proto();
102 Format(root->show_children, &root->formatted_str, root->mutable_proto()); in ShowInternal()183 node->mutable_proto()->mutable_tensor_value()); in PrintScope()
109 Format(root->show_children, &root->formatted_str, root->mutable_proto()); in ShowInternal()200 node->mutable_proto()->mutable_tensor_value()); in PrintGraph()
186 auto* pre_pb = root_->mutable_proto(); in ShowInternal()189 pre_pb->add_children()->Swap(show_node->mutable_proto()); in ShowInternal()
497 root->mutable_proto(), &call_ids); in ShowInternal()505 root->mutable_proto(), nullptr); in ShowInternal()
59 HloModuleProto* mutable_proto() { return &proto_; } in mutable_proto() function
363 HloModuleProto* module = computation.mutable_proto(); in Build()2621 HloModuleProto* module = computation.mutable_proto(); in BuildConstantSubGraph()