Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/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.cc53 xla::Shape TpuTransferManager::HostShapeToDeviceShape( in HostShapeToDeviceShape() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h54 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape() function
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc106 xla::Shape HostShapeToDeviceShape(const xla::Shape& host_shape) { in HostShapeToDeviceShape() function