Home
last modified time | relevance | path

Searched refs:as_graph_element (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py460 graph.as_graph_element("new_hidden1/Relu:0"), name="hidden1/Relu")
476 graph.as_graph_element("new_hidden2/Relu:0"), name="hidden2/Relu")
734 graph.as_graph_element("new_queue1/dequeue:0")
735 graph.as_graph_element("new_queue1/close")
882 self.assertEqual("/device:CPU:0", str(graph1.as_graph_element("a").device))
884 str(graph1.as_graph_element("b").device))
886 str(graph1.as_graph_element("matmul").device))
896 self.assertEqual("", str(graph2.as_graph_element("a").device))
897 self.assertEqual("", str(graph2.as_graph_element("b").device))
898 self.assertEqual("", str(graph2.as_graph_element("matmul").device))
[all …]
Dops.py3814 def as_graph_element(self, obj, allow_tensor=True, allow_operation=True): member in Graph
3976 return self.as_graph_element(name, allow_tensor=False, allow_operation=True)
4024 return self.as_graph_element(name, allow_tensor=True, allow_operation=False)
4940 c = self.as_graph_element(c)
Dmeta_graph.py887 col_op = graph.as_graph_element(
Dconvert_to_constants.py857 self._func.graph.as_graph_element(node.name + ":0"))
Dops_test.py3027 self.assertEqual(a, g.as_graph_element(ConvertibleObj()))
3029 g.as_graph_element(NonConvertibleObj())
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py104 return graph.as_graph_element(tensor_info.name)
278 flat_feeds = [self.graph.as_graph_element(t) for t in flat_feeds]
322 graph_element = self.graph.as_graph_element(fetch)
354 t = pruned_graph.as_graph_element(ti.name)
656 nest.map_structure(import_graph.as_graph_element, inputs),
657 nest.map_structure(import_graph.as_graph_element, outputs))
Dwrap_function_test.py135 return ops.get_default_graph().as_graph_element('output:0')
141 [], wrapped.graph.as_graph_element(v.initializer.name))
168 pruned = wrapped.prune([], wrapped.graph.as_graph_element('output:0'))
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2.py105 filename_tensor = wrapped.graph.as_graph_element(
112 wrapped.graph.as_graph_element(saver_def.restore_op_name)])
249 asset_feed_tensors.append(wrapped.graph.as_graph_element(tensor_name))
253 fetches=[init_anchor, wrapped.graph.as_graph_element(init_op)])
Dutils_impl.py216 return graph.as_graph_element(
Dload_v1_in_v2_test.py440 root.graph.as_graph_element("x:0")
/external/tensorflow/tensorflow/python/training/
Dqueue_runner_impl.py178 self._queue = g.as_graph_element(
180 self._enqueue_ops = [g.as_graph_element(
183 self._close_op = g.as_graph_element(ops.prepend_name_scope(
185 self._cancel_op = g.as_graph_element(ops.prepend_name_scope(
Dbasic_session_run_hooks.py1108 element = graph.as_graph_element(obj)
1110 element = graph.as_graph_element(obj + ":0")
1113 graph.as_graph_element(obj + ":1")
Dsaver_test.py2832 hidden1 = graph.as_graph_element("new_hidden1/Relu:0")
2833 weights1 = graph.as_graph_element("new_hidden1/weights:0")
2834 biases1 = graph.as_graph_element("new_hidden1/biases:0")
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_ops.py689 self._external_values[k] = g.as_graph_element(
697 g.as_graph_element(op)._set_control_flow_context(self)
880 self._pred = g.as_graph_element(
882 self._pivot = g.as_graph_element(
1549 self._maximum_iterations = g.as_graph_element(
1557 self._pivot_for_pred = g.as_graph_element(
1560 self._pivot_for_body = g.as_graph_element(
1564 self._pivot = g.as_graph_element(
1568 g.as_graph_element(ops.prepend_name_scope(exit_name, import_scope))
1573 g.as_graph_element(ops.prepend_name_scope(enter_name, import_scope))
[all …]
Dvariables.py1883 self._variable = g.as_graph_element(
1887 self._initializer_op = g.as_graph_element(
1893 self._initial_value = g.as_graph_element(
1905 self._snapshot = g.as_graph_element(
Dresource_variable_ops.py1893 self._handle = g.as_graph_element(
1899 self._initializer_op = g.as_graph_element(
1905 self._initial_value = g.as_graph_element(
1918 snapshot = g.as_graph_element(
Dgradients_test.py314 gradient = gradients.gradients(graph.as_graph_element(var), var)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-graph.pbtxt50 name: "as_graph_element"
Dtensorflow.__internal__.-func-graph.pbtxt107 name: "as_graph_element"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-graph.pbtxt50 name: "as_graph_element"
/external/tensorflow/tensorflow/python/client/
Dsession.py306 self._unique_fetches.append(ops.get_default_graph().as_graph_element(
1074 subfeed_t = self.graph.as_graph_element(
1130 subfeed_t = self.graph.as_graph_element(
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental_test.py282 return graph.as_graph_element(op_name)
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py799 col_op = dest_graph.as_graph_element(name)