Merge pull request #43290 from Ricardicus:master
PiperOrigin-RevId: 333465294 Change-Id: I52da1244e968d5e7fe6677fb59fa6a65946e8fef
This commit is contained in:
commit
5aeb5a0305
@ -64,7 +64,7 @@ class ClientSession {
|
||||
ClientSession(const Scope& scope, const string& target);
|
||||
|
||||
/// Same as above, but use the empty string ("") as the target specification.
|
||||
ClientSession(const Scope& scope);
|
||||
explicit ClientSession(const Scope& scope);
|
||||
|
||||
/// Create a new session, configuring it with `session_options`.
|
||||
ClientSession(const Scope& scope, const SessionOptions& session_options);
|
||||
|
@ -16,6 +16,8 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_CC_GRADIENTS_GRAD_TESTUTIL_H_
|
||||
#define TENSORFLOW_CC_GRADIENTS_GRAD_TESTUTIL_H_
|
||||
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/cc/framework/ops.h"
|
||||
#include "tensorflow/cc/framework/scope.h"
|
||||
|
||||
|
@ -16,6 +16,8 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_CC_OPS_CONST_OP_H_
|
||||
#define TENSORFLOW_CC_OPS_CONST_OP_H_
|
||||
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/cc/framework/ops.h"
|
||||
#include "tensorflow/cc/framework/scope.h"
|
||||
#include "tensorflow/core/graph/node_builder.h"
|
||||
|
@ -16,6 +16,9 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_CC_OPS_WHILE_LOOP_H_
|
||||
#define TENSORFLOW_CC_OPS_WHILE_LOOP_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/cc/framework/ops.h"
|
||||
#include "tensorflow/cc/framework/scope.h"
|
||||
|
||||
|
@ -16,6 +16,9 @@ limitations under the License.
|
||||
#ifndef TENSORFLOW_CC_PROFILER_PROFILER_H_
|
||||
#define TENSORFLOW_CC_PROFILER_PROFILER_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
||||
#include "tensorflow/core/framework/graph.pb.h"
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
#include "tensorflow/core/profiler/internal/tfprof_stats.h"
|
||||
@ -56,7 +59,7 @@ namespace tfprof {
|
||||
class Profiler {
|
||||
public:
|
||||
/// `graph` is the model's GraphDef.
|
||||
Profiler(const GraphDef& graph);
|
||||
explicit Profiler(const GraphDef& graph);
|
||||
|
||||
/// Adds tracing information `run_meta` to profiler. A `run_meta` is
|
||||
/// generated by a TensorFlow session run call. `step` is the key
|
||||
|
Loading…
Reference in New Issue
Block a user