Home
last modified time | relevance | path

Searched defs:HostShapeToDeviceShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executable.h57 Shape HostShapeToDeviceShape(const Shape& host_shape) override { in HostShapeToDeviceShape() function
Dtpu_op_executable.cc105 xla::Shape TpuOpExecutable::HostShapeToDeviceShape( in HostShapeToDeviceShape() function in tensorflow::TpuOpExecutable
Dtpu_transfer_manager.cc54 xla::Shape TpuTransferManager::HostShapeToDeviceShape( in HostShapeToDeviceShape() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_op_executable.cc105 xla::Shape TpuOpExecutable::HostShapeToDeviceShape( in HostShapeToDeviceShape() function in tensorflow::TpuOpExecutable
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h51 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape() function
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc108 xla::Shape HostShapeToDeviceShape(const xla::Shape& host_shape) { in HostShapeToDeviceShape() function