Method gpu

tensorflow/tensorflow/python/framework/ops.py:1057–1068 · view source on GitHub ↗

A copy of this Tensor with contents backed by memory on the GPU. Arguments: gpu_index: Identifies which GPU to place the contents on the returned Tensor in. Returns: A GPU-memory backed Tensor object initialized with the same contents as this Tensor.

(self, gpu_index=0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

copy_tensorMethod · 0.45

testMatMulGPUMethod · 0.45

testOpWithInputsOnDifferentDevicesMethod · 0.45

testOutputOnHostMemoryMethod · 0.45

testSilentCopyMethod · 0.45

testTensorPlacementMethod · 0.45

testCopyBetweenDevicesMethod · 0.45

testCopyBetweenDevicesAsyncMethod · 0.45

testNumpyForceCPUMethod · 0.45

testMatMulGPUMethod · 0.45

testOperationWithNoInputsRunsOnDeviceMethod · 0.45

benchmark_tf_multiply_GPUMethod · 0.45

Calls 2

_copyMethod · 0.95

contextMethod · 0.45

Tested by 15

copy_tensorMethod · 0.36

testMatMulGPUMethod · 0.36

testOpWithInputsOnDifferentDevicesMethod · 0.36

testOutputOnHostMemoryMethod · 0.36

testSilentCopyMethod · 0.36

testTensorPlacementMethod · 0.36

testCopyBetweenDevicesMethod · 0.36

testCopyBetweenDevicesAsyncMethod · 0.36

testNumpyForceCPUMethod · 0.36

testMatMulGPUMethod · 0.36

testOperationWithNoInputsRunsOnDeviceMethod · 0.36

benchmark_tf_multiply_GPUMethod · 0.36