Merge pull request #43290 from Ricardicus:master

PiperOrigin-RevId: 333465294
Change-Id: I52da1244e968d5e7fe6677fb59fa6a65946e8fef
This commit is contained in:
TensorFlower Gardener 2020-09-24 01:24:58 -07:00
commit 5aeb5a0305
5 changed files with 12 additions and 2 deletions

View File

@ -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);

View File

@ -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"

View File

@ -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"

View File

@ -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"

View File

@ -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