Home
last modified time | relevance | path

Searched defs:device_ptr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc403 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer()
Dpjrt_stream_executor_client.cc1076 void* device_ptr, const Shape& shape, PjRtDevice* device, in CreateViewOfDeviceBuffer()