Searched refs:collect_resource_usage_ (Results 1 – 2 of 2) sorted by relevance
644 : collect_resource_usage_(false), in Model()660 bool collect_resource_usage() const { return collect_resource_usage_; } in collect_resource_usage()796 std::atomic<bool> collect_resource_usage_; variable
1615 collect_resource_usage_ = in AddNode()1616 collect_resource_usage_ || node->has_tunable_parameters(); in AddNode()1937 model_proto->set_collect_resource_usage(collect_resource_usage_); in ToProto()1949 restored_model->collect_resource_usage_.store( in FromProto()1963 model_snapshot->collect_resource_usage_.store(collect_resource_usage_); in Save()