Home
last modified time | relevance | path

Searched refs:ReadDynamicShapes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h196 virtual Status ReadDynamicShapes(se::Stream* stream,
Dtransfer_manager.cc202 Status TransferManager::ReadDynamicShapes(se::Stream* stream, in ReadDynamicShapes() function in xla::TransferManager
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc147 node_context->backend()->transfer_manager()->ReadDynamicShapes( in AllocateOutputTuple()
Dxrt_execute_op.cc279 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in CreateOutputTuple()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc445 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in AllocateOutputTensors()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc469 TF_RETURN_IF_ERROR(transfer_manager->ReadDynamicShapes( in PopulateOutputs()