Searched refs:async_tensor (Results 1 – 1 of 1) sorted by relevance
384 AsyncValue* async_tensor = value_.get<TensorHandle>().GetAsyncTensor(); in DataType() local385 if (!async_tensor->IsAvailable()) { in DataType()386 context_.GetHostContext()->Await(FormRef(async_tensor)); in DataType()389 if (async_tensor->IsError()) { in DataType()391 << async_tensor->GetError().message; in DataType()394 assert(async_tensor->IsType<tensorflow::tfd::RuntimeFallbackTensor>()); in DataType()395 return async_tensor->get<tensorflow::tfd::RuntimeFallbackTensor>() in DataType()415 AsyncValue* async_tensor = value_.get<TensorHandle>().GetAsyncTensor(); in TensorHandleStatus() local416 if (!async_tensor->IsAvailable()) { in TensorHandleStatus()417 context_.GetHostContext()->Await(FormRef(async_tensor)); in TensorHandleStatus()[all …]