Searched defs:dht (Results 1 – 8 of 8) sorted by relevance
120 if (auto* dht = llvm::dyn_cast<DenseHostTensor>(&tensor)) { in TFRTTensorToTFTensor() local165 auto& dht = dht_ref.get(); in CreateTensorHandleFromTFTensor() local203 const auto& dht = llvm::cast<DenseHostTensor>(host_tensor); in CreateTFTensorFromTensorHandle() local
172 Argument<DenseHostTensor> dht, Argument<Chain> in_chain, in TfdMoveDHTToTFT()187 Result<DenseHostTensor> dht, in TfdConvertTFTToDHT()322 OwnedTFTensor MoveDHTToTFTensor(DenseHostTensor&& dht, HostContext* host) { in MoveDHTToTFTensor()343 Expected<DenseHostTensor> dht = in DecodeDenseAttrToTensorInterface() local1202 if (auto* dht = llvm::dyn_cast<tfrt::DenseHostTensor>(tensor)) { in ConvertTFRTTensorToTFTensorHandle() local
171 RuntimeFallbackTensor MoveDHTToRuntimeFallbackTensor(DenseHostTensor&& dht, in MoveDHTToRuntimeFallbackTensor()187 const DenseHostTensor& dht, HostContext* host) { in CopyRefDHTToRuntimeFallbackTensor()
164 auto dht = tfrt::MakeAvailableAsyncValueRef<DenseHostTensor>( in BatchFunctionFallback() local
338 auto& dht = tensor_.get<DenseHostTensor>(); in CanMove() local623 auto& dht = t.getValue(); in MakeScalarTensor() local761 AsyncValueRef<DenseHostTensor> dht = in CreateTensor() local815 if (auto* dht = llvm::dyn_cast<DenseHostTensor>(&tensor_av.get<Tensor>())) { in CreateLocalHandle() local829 if (auto* dht = in CreateLocalHandle() local891 if (auto* dht = llvm::dyn_cast<tfrt::DenseHostTensor>(&tensor)) { in TFTensorHandleFromInterface() local
49 tfrt::DenseHostTensor dht( in TEST() local
56 llvm::Expected<DenseHostTensor> dht = in DecodeDenseAttrToTfTensor() local
510 virtual bool SetDHT(const char __unused *dht) { return true; } in SetDHT()