Searched refs:ReadDynamicShapes (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 55 Status CpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::CpuTransferManager
60 return TransferManager::ReadDynamicShapes(stream, device_buffer, in ReadDynamicShapes()
|
| D | cpu_transfer_manager.h | 57 Status ReadDynamicShapes(se::Stream* stream, ShapedBuffer* device_buffer,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.h | 43 Status ReadDynamicShapes(se::Stream* stream, ShapedBuffer* device_buffer,
|
| D | gpu_transfer_manager.cc | 85 Status GpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::gpu::GpuTransferManager
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_transfer_manager.h | 91 Status ReadDynamicShapes(se::Stream* stream, xla::ShapedBuffer* device_buffer,
|
| D | tpu_transfer_manager.cc | 328 Status TpuTransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in tensorflow::tpu::TpuTransferManager
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transfer_manager.h | 194 virtual Status ReadDynamicShapes(se::Stream* stream,
|
| D | transfer_manager.cc | 197 Status TransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::TransferManager
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 147 node_context->backend()->transfer_manager()->ReadDynamicShapes( in AllocateOutputTuple()
|
| D | xrt_execute_op.cc | 279 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in CreateOutputTuple()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 444 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 521 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 620 transfer_manager->ReadDynamicShapes(stream, &shaped_buffer, &ret_shape)); in logical_on_device_shape()
|