Searched refs:tf_ops (Results 1 - 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
H A Dcustom_gradient.py23 from tensorflow.python.framework import ops as tf_ops namespace
52 name = "CustomGradient-%s" % tf_ops.uid()
53 args = [tf_ops.convert_to_tensor(x) for x in args]
58 @tf_ops.RegisterGradient(name)
66 with tf_ops.get_default_graph().gradient_override_map(
72 input_tensors = [tf_ops.convert_to_tensor(x) for x in args]
H A Dgraph_callable.py29 from tensorflow.python.framework import ops as tf_ops namespace
254 container = tf_ops.get_default_graph()._container # pylint: disable=protected-access
255 graph_key = tf_ops.get_default_graph()._graph_key # pylint: disable=protected-access
288 if not all(isinstance(x, tf_ops.Tensor) for x in outputs_list):
313 if isinstance(x, tf_ops.Tensor)]
316 func_def_outputs = [x for x in outputs_list if isinstance(x, tf_ops.Tensor)]
334 x for x in captured_outlist if isinstance(x, tf_ops.Tensor)]
/external/tensorflow/tensorflow/contrib/graph_editor/
H A Dutil.py25 from tensorflow.python.framework import ops as tf_ops namespace
96 if isinstance(obj, tf_ops.Tensor):
197 if isinstance(tops, tf_ops.Graph):
202 check_types = (tf_ops.Operation, tf_ops.Tensor)
235 if isinstance(ops, tf_ops.Graph):
246 check_types = None if ignore_ts else tf_ops.Operation
248 return [op for op in ops if isinstance(op, tf_ops.Operation)]
262 if not isinstance(graph, tf_ops.Graph):
284 if isinstance(ts, tf_ops
[all...]
H A Dtransform.py32 from tensorflow.python.framework import ops as tf_ops namespace
161 op_ = tf_ops.Operation(node_def_, info.graph_, [], output_types_,
201 if isinstance(top, tf_ops.Operation):
203 elif isinstance(top, tf_ops.Tensor):
419 if not isinstance(dst_graph, tf_ops.Graph):
570 if not isinstance(dst_graph, tf_ops.Graph):
612 if not isinstance(dst_graph, tf_ops.Graph):
653 graph = util.get_unique_graph(flatten_target_ts, check_types=(tf_ops.Tensor))
H A Dselect.py27 from tensorflow.python.framework import ops as tf_ops namespace
412 if isinstance(seed_ops[0], tf_ops.Tensor):
477 if isinstance(seed_ops[0], tf_ops.Tensor):
648 if graph is not None and not isinstance(graph, tf_ops.Graph):
714 if graph is not None and not isinstance(graph, tf_ops.Graph):
H A Dsubgraph.py30 from tensorflow.python.framework import ops as tf_ops namespace
622 if not isinstance(graph, tf_ops.Graph):
/external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
H A Dop_queue_test.py22 from tensorflow.python.framework import ops as tf_ops namespace
31 with tf_ops.Graph().as_default():
H A Dfisher_factors_test.py27 from tensorflow.python.framework import ops as tf_ops namespace
47 with tf_ops.Graph().as_default():
56 with tf_ops.Graph().as_default():
142 with tf_ops.Graph().as_default(), self.test_session() as sess:
153 with tf_ops.Graph().as_default(), self.test_session() as sess:
165 with tf_ops.Graph().as_default(), self.test_session() as sess:
186 with tf_ops.Graph().as_default():
192 with tf_ops.Graph().as_default():
199 with tf_ops.Graph().as_default():
207 with tf_ops
[all...]
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
H A Dop_queue.py22 from tensorflow.python.framework import ops as tf_ops namespace
45 op_names = tf_ops.convert_to_tensor(list(sorted(op.name for op in ops)))
H A Destimator.py27 from tensorflow.python.framework import ops as tf_ops namespace
64 with tf_ops.device(next(self._cycle)):
268 with tf_ops.name_scope(
278 with tf_ops.name_scope("create_inv_update_thunk"):
H A Dfisher_factors.py29 from tensorflow.python.framework import ops as tf_ops namespace
66 with tf_ops.colocate_with(op[0]):
69 with tf_ops.colocate_with(op):
191 elif isinstance(param, (tf_ops.Tensor, variables.Variable)):
560 if isinstance(x, tf_ops.IndexedSlices):
575 if isinstance(x, tf_ops.IndexedSlices):
586 if isinstance(x, tf_ops.IndexedSlices):
597 if isinstance(x, tf_ops.IndexedSlices):
676 if isinstance(x, tf_ops.IndexedSlices):
691 if isinstance(x, tf_ops
[all...]
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
H A Dsugar.py26 from tensorflow.python.framework import ops as tf_ops namespace
81 with tf_ops.name_scope(self._name, 'lt_reshape_encode',
121 with tf_ops.name_scope(self._name, 'lt_reshape_decode',
/external/tensorflow/tensorflow/contrib/graph_editor/tests/
H A Dmatch.py24 from tensorflow.python.framework import ops as tf_ops namespace
77 elif isinstance(elem, tf_ops.Operation):
88 if not isinstance(op, tf_ops.Operation):

Completed in 2018 milliseconds