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.cc106 Shape TpuExecutable::HostShapeToDeviceShape(const Shape& host_shape) { in HostShapeToDeviceShape() function in xla::TpuExecutable
Dtpu_transfer_manager.cc52 xla::Shape TpuTransferManager::HostShapeToDeviceShape( in HostShapeToDeviceShape() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h53 virtual Shape HostShapeToDeviceShape(const Shape& host_shape) const { in HostShapeToDeviceShape() function
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc198 Shape HostShapeToDeviceShape(const Shape& host_shape) override { in HostShapeToDeviceShape() function in xla::__anoncef202870111::TpuExecutable
Dtpu_execute.cc106 xla::Shape HostShapeToDeviceShape(const xla::Shape& host_shape) { in HostShapeToDeviceShape() function