| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| checkpoint_utils_test.py | 118 # New graph and session.
119 with ops.Graph().as_default() as g:
120 with self.session(graph=g) as session:
144 # Check that tensors are not explicitly in the graph.
145 self.assertLess(len(str(session.graph.as_graph_def())), 27000)
152 with ops.Graph().as_default() as g:
161 with self.session(graph=g) as session:
171 # New graph and session.
172 with ops.Graph().as_default() as g:
173 with self.session(graph=g) as session
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session_with_tracking_alloc_test.cc | 25 #include "tensorflow/core/framework/graph.pb.h"
30 #include "tensorflow/core/graph/costmodel.h"
31 #include "tensorflow/core/graph/graph.h"
32 #include "tensorflow/core/graph/testlib.h"
49 Graph graph(OpRegistry::Global());
53 Node* a = test::graph::Constant(&graph, a_tensor);
58 Node* x = test::graph::Constant(&graph, x_tensor)
[all...] |
| lower_if_op_test.cc | 30 #include "tensorflow/core/graph/graph_constructor.h"
31 #include "tensorflow/core/graph/graph_def_builder.h"
32 #include "tensorflow/core/graph/graph_def_builder_util.h"
46 Status Rewrite(std::unique_ptr<Graph>* graph) {
47 FunctionLibraryDefinition flib_def((*graph)->flib_def());
49 opt_options.graph = graph;
56 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()))
[all...] |
| lower_while_op_test.cc | 28 #include "tensorflow/core/graph/graph_constructor.h"
29 #include "tensorflow/core/graph/graph_def_builder.h"
30 #include "tensorflow/core/graph/graph_def_builder_util.h"
38 Status Rewrite(std::unique_ptr<Graph>* graph) {
39 FunctionLibraryDefinition flib_def((*graph)->flib_def());
41 opt_options.graph = graph;
48 std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global()))
[all...] |
| lower_if_op.cc | 20 #include "tensorflow/core/graph/graph.h"
21 #include "tensorflow/core/graph/node_builder.h"
39 // `graph`. The functions should be available in `flib`.
42 Graph* graph);
63 // (name_), infix and a suffix to ensure it is unique within the graph.
93 Graph* graph_;
104 const FunctionLibraryDefinition& flib, Graph* graph)
[all...] |
| lower_while_op.cc | 21 #include "tensorflow/core/graph/graph.h"
22 #include "tensorflow/core/graph/node_builder.h"
34 // Input graph:
38 // Output graph(top to down flow):
57 Graph* graph, const FunctionLibraryDefinition& flib) {
59 parallel_iterations, graph, flib);
66 // the given graph.
69 Graph* graph, const FunctionLibraryDefinition& flib)
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view.cc | 27 #include "tensorflow/core/framework/graph.pb.h"
29 #include "tensorflow/core/graph/graph.h"
30 #include "tensorflow/core/graph/tensor_id.h"
44 return tensor_id.index() >= Graph::kControlSlot;
48 return tensor_id.index() > Graph::kControlSlot;
52 return tensor_id.index() == Graph::kControlSlot;
56 return port.port_id == Graph::kControlSlot;
61 bool IsIdentityConsumingSwitch(const MutableGraphView& graph,
70 NodeDef* input_node = graph.GetNode(tensor_id.node())
[all...] |
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| base_test.py | 37 """Create a graph containing single segment."""
38 # TODO(aaroey): test graph with different dtypes.
43 g = ops.Graph()
87 """Create a graph containing multiple segment."""
88 # TODO(aaroey): test graph with different dtypes.
93 g = ops.Graph()
146 # the graph and breaks the conversion check.
153 """Create a graph containing two segment."""
157 g = ops.Graph()
202 """Create a graph containing multiple segment.""
[all...] |
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph_test.py | 15 """Tests the graph freezing tool."""
58 # We'll create an input graph that has a single variable containing 1.0,
60 with ops.Graph().as_default():
74 graph_io.write_graph(sess.graph, self.get_temp_dir(), input_graph_name)
76 # We save out the graph to disk, and then call the const conversion
102 # Now we make sure the variable is now a constant, and that the graph still
104 with ops.Graph().as_default():
116 output_node = sess.graph.get_tensor_by_name("output_node:0")
129 with ops.Graph().as_default():
175 with ops.Graph().as_default()
[all...] |
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops_test.cc | 29 #include "tensorflow/core/graph/graph.h"
30 #include "tensorflow/core/graph/node_builder.h"
31 #include "tensorflow/core/graph/testlib.h"
325 Graph g{OpRegistry::Global()};
326 Node* encoded = test::graph::Constant(&g, encoded_tensor);
327 Node* shape = test::graph::Constant(&g, shape_tensor);
328 Node* cdf = test::graph::Constant(&g, cdf_tensor);
401 // This function creates RangeEncode graph with CDF built from a separate data
403 Graph* CreateRangeEncodeFullBroadcastGraph(const TensorShape& shape
[all...] |
| /external/tensorflow/tensorflow/python/training/ |
| checkpoint_utils_test.py | 120 # New graph and session.
121 with ops.Graph().as_default() as g:
122 with self.session(graph=g) as session:
146 # Check that tensors are not explicitly in the graph.
147 self.assertLess(len(str(session.graph.as_graph_def())), 29000)
154 # New graph and session.
155 with ops.Graph().as_default() as g:
156 with self.session(graph=g) as session:
184 with ops.Graph().as_default() as g:
193 with self.session(graph=g) as session
[all...] |
| saver_test.py | 87 with self.session(graph=ops_lib.Graph()) as sess:
88 # Build a graph with 2 parameter nodes, and Save and
118 with self.session(graph=ops_lib.Graph()) as sess:
138 # Build another graph with 2 nodes, initialized
140 with self.session(graph=ops_lib.Graph()) as sess:
196 with ops_lib.Graph().as_default() as g:
223 # Save from graph mode and restore from eager mode
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| convert_nodes.h | 30 #include "tensorflow/core/framework/graph.pb.h"
31 #include "tensorflow/core/graph/graph.h"
71 outside_port(Graph::kControlSlot),
74 inside_port(Graph::kControlSlot),
76 port_number(Graph::kControlSlot) {}
78 bool is_control_edge() const { return port_number == Graph::kControlSlot; }
122 // Constructs a graphdef from the segment in the given graph. Adds placeholder
125 // inserted in order to correctly get sizes from the original graph.
136 const Graph* graph, const grappler::GraphProperties& graph_properties
[all...] |
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| sampling_ops_test.py | 149 with ops.Graph().as_default():
168 with ops.Graph().as_default():
187 with ops.Graph().as_default():
205 with ops.Graph().as_default():
273 with ops.Graph().as_default():
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column_v2_test.py | 421 with ops.Graph().as_default():
435 with ops.Graph().as_default():
554 with ops.Graph().as_default():
569 with ops.Graph().as_default():
589 with ops.Graph().as_default():
609 with ops.Graph().as_default():
627 with ops.Graph().as_default():
667 with ops.Graph().as_default():
695 with ops.Graph().as_default():
724 with ops.Graph().as_default()
[all...] |
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| receptive_field_test.py | 39 The graph corresponds to the example from the second figure in
43 g: Tensorflow graph object (Graph proto).
45 g = ops.Graph()
64 The graph corresponds to a variation to the example from the second figure in
71 g: Tensorflow graph object (Graph proto).
73 g = ops.Graph()
92 The graph corresponds to the example from the first figure in
96 g: Tensorflow graph object (Graph proto)
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| mkl_layout_pass_test.cc | 18 #include "tensorflow/core/graph/mkl_layout_pass.h"
19 #include "tensorflow/core/graph/mkl_graph_util.h"
26 #include "tensorflow/core/graph/graph.h"
27 #include "tensorflow/core/graph/graph_constructor.h"
28 #include "tensorflow/core/graph/testlib.h"
40 // NOTE: Unit tests in this file rely on a topological sorted graph for
41 // printing. But since sibling nodes of a node in the topologically sorted graph
50 static void InitGraph(const string& s, Graph* graph,
3297 Graph* graph = new Graph(OpRegistry::Global()); local
[all...] |
| gradients.cc | 27 #include "tensorflow/core/graph/algorithm.h"
28 #include "tensorflow/core/graph/gradients.h"
29 #include "tensorflow/core/graph/graph_constructor.h"
30 #include "tensorflow/core/graph/optimizer_cse.h"
65 static Node* AddZerosLike(Graph* g, NodeOut input) {
101 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) {
157 Graph* graph);
166 Graph* graph_; // Not owned.
210 std::vector<NodeOut>* x_grad_node_outputs, Graph* graph
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| vectorization_utils.cc | 32 #include "tensorflow/core/graph/node_builder.h"
52 Graph* graph) {
62 graph->AddEdge(new_src.first, new_src.second, edge->dst(),
64 graph->RemoveEdge(edge);
90 Node* ret_node = map_defun_fn->graph->AddNode(ret_node_def, &s);
93 map_defun_fn->graph->AddEdge(output.first, output.second, ret_node, 0);
101 void RemoveMapDefunOutput(int output_position, Graph* outer_scope,
109 // Modify map_defun_fn's signature and remove the output node from its graph
110 map_defun_fn->graph->RemoveNode(map_defun_fn->ret_nodes[output_position])
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| optimizers_test.py | 68 with ops.Graph().as_default() as g:
69 with self.session(graph=g) as session:
84 with ops.Graph().as_default() as g:
85 with self.session(graph=g) as session:
98 with ops.Graph().as_default() as g:
99 with self.session(graph=g):
106 with ops.Graph().as_default() as g, self.session(graph=g):
114 with ops.Graph().as_default() as g, self.session(graph=g)
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_ops_pass.cc | 38 #include "tensorflow/core/graph/algorithm.h"
39 #include "tensorflow/core/graph/graph.h"
40 #include "tensorflow/core/graph/graph_constructor.h"
48 void MoveOutgoingEdges(Graph* g, Node* old_node, Node* new_node) {
64 scope.graph()->AddControlEdge(control, data.node());
100 s.graph()->RemoveEdge(e);
101 s.graph()->AddEdge(merged_output.node(), merged_output.index(), dst,
124 s.graph()->AddControlEdge(new_node, s.graph()->sink_node())
402 Graph* graph = options.graph->get(); local
[all...] |
| encapsulate_xla_computations_pass.cc | 43 bool IsCpuGpuCompile(const Graph* graph) {
44 for (Node* n : graph->nodes()) {
62 // Checks if a graph node is marked to be a guaranteed constant.
114 std::unique_ptr<Graph>* graph_ptr,
118 Graph* graph = graph_ptr->get(); local
126 for (Node* n : graph->nodes()) {
188 graph->ToGraphDef(&gdef);
211 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def)
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_cond.cc | 33 #include "tensorflow/core/graph/algorithm.h"
34 #include "tensorflow/core/graph/control_flow.h"
35 #include "tensorflow/core/graph/node_builder.h"
157 StateMap::StateMap(Graph* graph) {
158 node_to_condid_map_.resize(graph->num_node_ids());
159 node_to_ancestorid_map_.resize(graph->num_node_ids());
285 FunctionalizeCond::FunctionalizeCond(Graph* graph,
287 : state_map_(graph), library_(library), graph_(graph) {
[all...] |
| /external/tensorflow/tensorflow/core/debug/ |
| debug_graph_utils.cc | 22 #include "tensorflow/core/graph/node_builder.h"
49 const protobuf::RepeatedPtrField<DebugTensorWatch>& watches, Graph* graph,
106 for (Node* src_node : graph->nodes()) {
145 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node,
151 graph, device_type, memory_type == HOST_MEMORY, src_node->name(),
162 graph->AddEdge(src_node, src_output_slot, copy_node, 0);
171 graph, *device, copy_node->name(), src_dt, tensor_name,
174 graph->AddEdge(copy_node, 0, debug_node, 0);
199 graph->AddEdge(copy_node, 0, edge->dst(), edge->dst_input())
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| example_parsing_ops_test.cc | 27 #include "tensorflow/core/graph/graph.h"
28 #include "tensorflow/core/graph/node_builder.h"
145 static Graph* ParseExample(int batch_size, int num_keys, int feature_size) {
146 Graph* g = new Graph(OpRegistry::Global());
162 dense_keys.emplace_back(test::graph::Constant(g, key));
163 dense_defaults.emplace_back(test::graph::Constant(
168 dense_keys.emplace_back(test::graph::Constant(g, key));
170 test::graph::Constant(g, opt.filler.make_dense_default(1)))
[all...] |
