Home
last modified time | relevance | path

Searched defs:device_to_device_stream (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_context.h35 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream) in PluggableDeviceContext()
47 se::Stream* device_to_device_stream(int index) const { in device_to_device_stream() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgpu_device_context.h38 gtl::InlinedVector<se::Stream*, 4> device_to_device_stream, in GPUDeviceContext()
59 se::Stream* device_to_device_stream(int index) const { in device_to_device_stream() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device_context.h80 se::Stream* device_to_device_stream(int index) const { in device_to_device_stream() function