diff options
Diffstat (limited to 'tensorflow/python/framework/ops.py')
-rw-r--r-- | tensorflow/python/framework/ops.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/python/framework/ops.py b/tensorflow/python/framework/ops.py index 5948e59824..1b7b9eea12 100644 --- a/tensorflow/python/framework/ops.py +++ b/tensorflow/python/framework/ops.py @@ -1632,7 +1632,7 @@ class Operation(object): def _create_c_op(self, graph, node_def, inputs, control_inputs): """Creates a TF_Operation. - Arguments: + Args: graph: a `Graph`. node_def: `node_def_pb2.NodeDef` for the operation to create. inputs: A list of `Tensor`s (corresponding to scalar inputs) and lists of @@ -1677,7 +1677,7 @@ class Operation(object): def _reconstruct_sequence_inputs(self, op_def, inputs, attrs): """Regroups a flat list of input tensors into scalar and sequence inputs. - Arguments: + Args: op_def: The `op_def_pb2.OpDef` (for knowing the input types) inputs: a list of input `Tensor`s to the op. attrs: mapping from attr name to `attr_value_pb2.AttrValue` (these define @@ -3763,7 +3763,7 @@ class Graph(object): For example: ```python - with g.device('/gpu:0'): + with g.device('/device:GPU:0'): # All operations constructed in this context will be placed # on GPU 0. with g.device(None): @@ -3773,7 +3773,7 @@ class Graph(object): # Defines a function from `Operation` to device string. def matmul_on_gpu(n): if n.type == "MatMul": - return "/gpu:0" + return "/device:GPU:0" else: return "/cpu:0" |