| /external/tensorflow/tensorflow/python/kernel_tests/ |
| xent_op_test.py | 205 # there is no `BatchMatMul` op in the graph.
207 op.op_def.name for op in sess.graph.get_operations() if op.op_def
258 # (it is equivalent to being `BatchMatMul` op in the graph because of implementation of xentropy grad)
260 op.op_def.name for op in sess.graph.get_operations() if op.op_def
307 with ops.Graph().as_default():
332 with ops.Graph().as_default():
|
| /external/tensorflow/tensorflow/python/training/ |
| queue_runner_test.py | 101 self.assertTrue("Operation not in the graph" in str(exceptions[0]))
102 self.assertTrue("Operation not in the graph" in str(exceptions[1]))
165 with self.assertRaisesRegexp(ValueError, "Operation not in the graph"):
234 self.assertTrue("Operation not in the graph" in str(exceptions[0]))
241 self.assertTrue("Operation not in the graph" in str(exceptions[0]))
299 graph = ops.Graph()
300 with graph.as_default():
308 with self.session(graph=graph) as sess
[all...] |
| /external/v8/src/compiler/ |
| graph-assembler.h | 8 #include "src/compiler/js-graph.h"
17 class Graph;
275 Graph* graph() const { return jsgraph_->graph(); } function in class:v8::internal::compiler::GraphAssembler
305 label->control_ = graph()->NewNode(common()->Loop(2), current_control_,
307 label->effect_ = graph()->NewNode(common()->EffectPhi(2), current_effect_,
309 Node* terminate = graph()->NewNode(common()->Terminate(), label->effect_,
311 NodeProperties::MergeControlToEnd(graph(), common(), terminate);
313 label->bindings_[i] = graph()->NewNode
[all...] |
| js-inlining.cc | 10 #include "src/compiler/bytecode-graph-builder.h"
13 #include "src/compiler/graph-reducer.h"
156 Node* on_success = graph()->NewNode(common()->IfSuccess(), subcall);
160 graph()->NewNode(common()->IfException(), subcall, subcall);
167 graph()->NewNode(common()->Merge(subcall_count), subcall_count,
172 Node* value_output = graph()->NewNode(
176 graph()->NewNode(common()->EffectPhi(subcall_count),
199 NodeProperties::MergeControlToEnd(graph(), common(), input);
200 Revisit(graph()->end());
214 Node* control_output = graph()->NewNode(common()->Merge(input_count)
637 Graph* JSInliner::graph() const { return jsgraph()->graph(); } function in class:v8::internal::compiler::JSInliner
[all...] |
| loop-analysis.cc | 7 #include "src/compiler/graph.h"
46 // This loop finding algorithm works by traversing the graph in two directions,
54 LoopFinderImpl(Graph* graph, LoopTree* loop_tree, Zone* zone)
56 end_(graph->end()),
58 queued_(graph, 2),
59 info_(graph->NodeCount(), {nullptr, nullptr}, zone),
61 loop_num_(graph->NodeCount(), -1, zone),
516 LoopTree* LoopFinder::BuildLoopTree(Graph* graph, Zone* zone)
[all...] |
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| make_test_graphs.py | 130 """A more complex graph, including splits."""
172 """Build a graph using build_graph and write it out."""
173 g = ops.Graph()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| deadness_analysis_test.cc | 30 #include "tensorflow/core/graph/algorithm.h"
31 #include "tensorflow/core/graph/graph_constructor.h"
32 #include "tensorflow/core/graph/graph_def_builder.h"
33 #include "tensorflow/core/graph/graph_def_builder_util.h"
43 Status AnalyzeDeadness(Graph* graph,
45 FixupSourceAndSinkEdges(graph);
46 return DeadnessAnalysis::Run(*graph, result);
57 return {o.node()->name(), Graph::kControlSlot};
60 void VLogGraphIfAsked(const Graph& graph)
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| graph_compiler.cc | 41 #include "tensorflow/core/graph/algorithm.h"
42 #include "tensorflow/core/graph/graph_constructor.h"
43 #include "tensorflow/core/graph/node_builder.h"
44 #include "tensorflow/core/graph/validate.h"
55 Status PrepareArguments(XlaOpKernelContext* ctx, Graph* graph,
62 *graph, &arg_must_be_compile_time_constant,
107 // Check that the graph has no illegal cycles.
108 TF_RETURN_IF_ERROR(graph::ValidateGraphHasNoCycle(*graph_));
149 // tensor_inputs_ is a buffer reused across graph traversal. We clean up an
247 auto graph = compiler->GetGraph(fbody); local
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| gmm_ops_test.py | 120 graph = ops.Graph()
121 with graph.as_default() as g:
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| export.py | 69 def _export_graph(graph, saver, checkpoint_path, export_dir,
72 """Exports graph via session_bundle, by creating a Session."""
73 with graph.as_default():
304 with ops.Graph().as_default() as g:
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| model_average_optimizer_test.py | 66 graph = ops.Graph()
68 with graph.as_default():
107 graphs.append(graph)
|
| moving_average_optimizer_test.py | 74 with self.session(graph=ops.Graph()) as sess:
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| receptive_field.py | 158 """Computes receptive field (RF) parameters from a Graph or GraphDef object.
167 graph_def: Graph or GraphDef object.
168 input_node: Name of the input node or Tensor object from graph.
169 output_node: Name of the output node or Tensor object from graph.
198 # Convert a graph to graph_def if necessary.
199 if isinstance(graph_def, framework_ops.Graph):
210 # Computes order of computation for a given graph.
339 # to make sure that graph is aligned. This alignment check is skipped
345 "Graph is not aligned since effective stride from different "
349 "Graph is not aligned since effective stride from different
[all...] |
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| bundle_shim.cc | 23 #include "tensorflow/core/graph/graph_constructor.h"
118 // graph def. Use the default key if it isn't already in use.
308 Graph graph(OpRegistry::Global());
310 ConvertGraphDefToGraph(opts, meta_graph_def.graph_def(), &graph));
311 for (Node* node : graph.nodes()) {
343 // Copy the meta graph def from the SessionBundle to the SavedModelBundle.
|
| session_bundle_test.py | 123 self.assertTrue("Expected meta graph file missing" in str(cm.exception))
155 """Test the case where there are no variables in the graph."""
162 # Create a simple graph with a variable, then convert variables to
163 # constants and export the graph.
164 with ops.Graph().as_default() as g:
170 with self.session(graph=g) as session:
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| inception_v1_test.py | 79 with ops.Graph().as_default():
|
| /external/tensorflow/tensorflow/core/framework/ |
| graph_to_functiondef.cc | 26 #include "tensorflow/core/graph/graph.h"
113 // Graph to FunctionDef conversion. This code is closely modeled on the Python
116 Status GraphToFunctionDef(const Graph& graph, const string& name,
124 for (Node const* node : graph.op_nodes()) {
170 // Reset input names based on graph rather than the NodeDef.
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| graph_utils_test.cc | 19 #include "tensorflow/core/graph/node_builder.h"
42 MutableGraphView graph(&graph_def);
43 NodeDef* bool_node = AddScalarConstNode<bool>(true, &graph);
44 EXPECT_TRUE(ContainsGraphNodeWithName(bool_node->name(), *graph.graph()));
50 MutableGraphView graph(&graph_def);
51 NodeDef* double_node = AddScalarConstNode<double>(3.14, &graph);
52 EXPECT_TRUE(ContainsGraphNodeWithName(double_node->name(), *graph.graph()));
58 MutableGraphView graph(&graph_def)
[all...] |
| /external/tensorflow/tensorflow/go/ |
| session_test.go | 25 func createTestGraph(t *testing.T, dt DataType) (*Graph, Output, Output) {
54 graph, inp, out := createTestGraph(t, t1.DataType())
55 s, err := NewSession(graph, &SessionOptions{})
125 // Construct the graph:
170 graph, inp, out := createTestGraph(t, tensor.DataType())
171 s, err := NewSession(graph, &SessionOptions{})
189 // Create a graph: a + 2 + 3 + b.
192 // The 'op' package can be used to make graph construction code
268 graph := NewGraph()
269 c, err := Const(graph, "Const", int32(14)
[all...] |
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| make_batched_features_dataset_test.py | 192 with ops.Graph().as_default():
|
| /external/tensorflow/tensorflow/python/distribute/cluster_resolver/ |
| cluster_resolver.py | 59 with ops.Graph().as_default():
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| sequence_feature_column_test.py | 170 with ops.Graph().as_default():
[all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| optimizers_test.py | 122 # nodes in the graph seemed to affect the initialization seed, and hence
175 graph = ops.Graph()
176 with graph.as_default():
180 graph_weak = weakref.ref(graph)
181 del graph, optimizer
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| random_gamma_test.py | 46 with self.session(use_gpu=use_gpu, graph=ops.Graph()) as sess:
|
| /external/tensorflow/tensorflow/python/ops/ |
| init_ops_test.py | 182 with ops.Graph().as_default() as g:
196 with session.Session(config=config, graph=g) as sess:
|
