Skip to content

Add #device to place ops on specific named devices #67

@chrhansen

Description

@chrhansen

Used in "Basic Usage" here: https://www.tensorflow.org/versions/r0.10/get_started/basic_usage.html e.g.:

with tf.device('/gpu:2'):
  tf.constant([1.0, 2.0, 3.0], shape=[3], name='a')

Could be something like (using default-graph if not explicitly specified) this in Ruby:

Tensorflow.device('/gpu:2') do
  graph.constant([1.0, 2.0, 3.0], shape: [3], name : 'a')
end

Python API: https://www.tensorflow.org/versions/r0.10/how_tos/using_gpu/index.html#supported-devices

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions