Home
last modified time | relevance | path

Searched defs:ReadDynamicShapes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_transfer_manager.cc55 Status CpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::CpuTransferManager
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_transfer_manager.cc85 Status GpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::gpu::GpuTransferManager
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_transfer_manager.cc328 Status TpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in tensorflow::tpu::TpuTransferManager
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtransfer_manager.cc197 Status TransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::TransferManager