Home
last modified time | relevance | path

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

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