Searched refs:c_host_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.cc | 54 XLA_Shape c_host_shape; in HostShapeToDeviceShape() local
57 ApiConverter::ToC(host_shape, &c_host_shape); in HostShapeToDeviceShape()
60 manager_, &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
62 ApiConverter::Free(&c_host_shape); in HostShapeToDeviceShape()
226 XLA_Shape c_host_shape; in ChooseCompactLayoutForShape() local
227 ApiConverter::ToC(host_shape, &c_host_shape); in ChooseCompactLayoutForShape()
231 manager_, &c_host_shape, &c_output, status.c_status); in ChooseCompactLayoutForShape()
233 ApiConverter::Free(&c_host_shape); in ChooseCompactLayoutForShape()
|
| D | tpu_executable.cc | 107 XLA_Shape c_host_shape; in HostShapeToDeviceShape() local
109 ApiConverter::ToC(host_shape, &c_host_shape); in HostShapeToDeviceShape()
111 &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
113 ApiConverter::Free(&c_host_shape); in HostShapeToDeviceShape()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 107 XLA_Shape c_host_shape; in HostShapeToDeviceShape() local
109 ApiConverter::ToC(host_shape, &c_host_shape); in HostShapeToDeviceShape()
111 &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
113 ApiConverter::Free(&c_host_shape); in HostShapeToDeviceShape()
|