Searched refs:mutable_proto (Results 1 – 9 of 9) sorted by relevance
26 mutable_proto()->set_name(name()); in ReInit()27 mutable_proto()->clear_devices(); in ReInit()29 mutable_proto()->add_devices(node->canonical_device()); in ReInit()31 mutable_proto()->set_run_count(node->run_count(step)); in ReInit()32 mutable_proto()->set_exec_micros(node->exec_micros(step)); in ReInit()33 mutable_proto()->set_accelerator_exec_micros( in ReInit()35 mutable_proto()->set_cpu_exec_micros(node->cpu_exec_micros(step)); in ReInit()37 mutable_proto()->set_requested_bytes(node->requested_bytes(step)); in ReInit()38 mutable_proto()->set_peak_bytes(node->peak_bytes(step)); in ReInit()39 mutable_proto()->set_residual_bytes(node->residual_bytes(step)); in ReInit()[all …]
47 GraphNodeProto* mutable_proto();93 MultiGraphNodeProto* mutable_proto();
101 Format(root->show_children, &root->formatted_str, root->mutable_proto()); in ShowInternal()181 node->mutable_proto()->mutable_tensor_value()); in PrintScope()
110 Format(root->show_children, &root->formatted_str, root->mutable_proto()); in ShowInternal()200 node->mutable_proto()->mutable_tensor_value()); in PrintGraph()
183 auto* pre_pb = root_->mutable_proto(); in ShowInternal()186 pre_pb->add_children()->Swap(show_node->mutable_proto()); in ShowInternal()
510 root->mutable_proto(), &call_ids); in ShowInternal()518 root->mutable_proto(), nullptr); in ShowInternal()
51 HloModuleProto* mutable_proto() { return &proto_; } in mutable_proto() function
558 HloModuleProto* module = computation.mutable_proto(); in Build()3809 HloModuleProto* module = computation.mutable_proto(); in BuildConstantSubGraph()
109 if (!header.mutable_proto()->ParseFromArray(executable.bytes, in SerializeCacheEntryToBufferSlices()