Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_values.cc272 py::object device_array = in DevicePut() local
433 py::object device_array = in PyArgSignatureOfValue() local
Djax_jit.cc563 py::object device_array; in ComputeSignature() member
567 py::object device_array(xla_module.attr("_DeviceArray")); in ComputeSignature() local
916 static const auto* device_array = in Call() local