Home
last modified time | relevance | path

Searched refs:c_device_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable.cc108 XLA_Shape c_device_shape; in HostShapeToDeviceShape() local
111 &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
112 Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()
114 ApiConverter::Free(&c_device_shape); in HostShapeToDeviceShape()
Dtpu_transfer_manager.cc55 XLA_Shape c_device_shape; in HostShapeToDeviceShape() local
60 manager_, &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
61 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()
63 ApiConverter::Free(&c_device_shape); in HostShapeToDeviceShape()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc108 XLA_Shape c_device_shape; in HostShapeToDeviceShape() local
111 &c_host_shape, &c_device_shape); in HostShapeToDeviceShape()
112 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape()
114 ApiConverter::Free(&c_device_shape); in HostShapeToDeviceShape()