Searched refs:StatusHelper (Results 1 – 14 of 14) sorted by relevance
41 StatusHelper status; in Init()64 StatusHelper status; in BlockHostUntilDone()71 StatusHelper status; in BlockUntilDoneOrFailed()78 StatusHelper status; in GetStatus()142 StatusHelper status; in RecordEvent()152 StatusHelper status; in WaitForEvent()248 StatusHelper status; in WaitForInfeedReady()255 StatusHelper status; in WaitForOutfeedReady()263 StatusHelper status; in DequeueOutfeed()272 StatusHelper status; in EnqueueInfeed()[all …]
31 StatusHelper status; in Create()49 StatusHelper status; in StopChipHeartbeats()56 StatusHelper status; in CloseTpuHost()63 StatusHelper status; in Initialize()
23 class StatusHelper {25 StatusHelper() in StatusHelper() function28 ~StatusHelper() { in ~StatusHelper()
71 StatusHelper status; in TransferLiteralToDeviceAsync()92 StatusHelper status; in TransferLiteralToInfeed()108 StatusHelper status; in TransferBuffersToInfeed()132 StatusHelper status; in TransferLiteralFromOutfeed()152 StatusHelper status; in ResetDevices()179 done(StatusHelper::FromC(overall_status)); in TransferFinished()229 StatusHelper status; in ChooseCompactLayoutForShape()283 StatusHelper status; in WriteSingleTupleIndexTable()305 StatusHelper status; in LinearizeToBuffers()
48 StatusHelper status; in EnqueueTransferHostToDevice()58 StatusHelper status; in EnqueueTransferDeviceToHost()68 StatusHelper status; in EnqueueOnTpuDeviceSendRecvLocal()
45 StatusHelper status; in Initialize()96 StatusHelper status; in GetUncachedExecutor()
78 StatusHelper status; in LoadProgramAndEnqueueToStream()
62 StatusHelper status; in TpuTracer()74 StatusHelper status; in Start()84 StatusHelper status; in Stop()99 StatusHelper status; in CollectData()
40 StatusHelper status; in ConstructExecutableInfo()56 StatusHelper status; in ConstructHostTransferInfo()72 StatusHelper status; in ConstructHloMetadata()143 StatusHelper status; in UnloadAndDestroyPrograms()257 StatusHelper status; in CompileAndBuild()291 StatusHelper status; in CompileAndBuild()335 StatusHelper status; in DeserializeFromRpcResponseProtos()354 StatusHelper status; in SerializeExecutable()364 StatusHelper status; in SerializeCompilerMetadata()
45 StatusHelper status; in AssignDevices()60 StatusHelper status; in AssignLocalDevices()
133 StatusHelper status; in ExecuteAsyncOnStream()231 StatusHelper status; in RunHloPasses()260 StatusHelper status; in RunBackend()314 StatusHelper status; in Compile()
242 StatusHelper status; in UpdateDynamicInputs()
24 TEST(StatusHelper, TestStatusHelper) { in TEST() argument
61 StatusHelper status; in TpuShapeRepresentation()90 StatusHelper status; in TpuPaddedShapeFn()