/external/tensorflow/tensorflow/python/eager/ |
H A D | custom_gradient.py | 23 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 D | graph_callable.py | 29 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 D | util.py | 25 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 D | transform.py | 32 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 D | select.py | 27 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 D | subgraph.py | 30 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 D | op_queue_test.py | 22 from tensorflow.python.framework import ops as tf_ops namespace 31 with tf_ops.Graph().as_default():
|
H A D | fisher_factors_test.py | 27 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 D | op_queue.py | 22 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 D | estimator.py | 27 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 D | fisher_factors.py | 29 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 D | sugar.py | 26 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 D | match.py | 24 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):
|