Searched defs:tensor_ref (Results 1 – 10 of 10) sorted by relevance
44 TensorRef<BHWC> tensor_ref; in TEST() local60 TensorRef<BHWC> tensor_ref; in TEST() local77 TensorRef<BHWC> tensor_ref; in TEST() local95 TensorRef<BHWC> tensor_ref; in TEST() local109 TensorRef<BHWC> tensor_ref; in TEST() local122 TensorRef<BHWC> tensor_ref; in TEST() local
157 TensorRef<BHWC>* tensor_ref) { in ConvertTfLiteTensorToTensorRef()
37 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
33 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
35 absl::Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref, in CreatePHWC4BufferFromTensorRef()
160 for (auto tensor_ref : scratch_tensor_refs_) { in ~ROCmSolver() local
194 for (const auto& tensor_ref : scratch_tensor_refs_) { in ~CudaSolver() local
270 TensorReference tensor_ref(*tensor); // This will call buf_->Ref() in TFE_HandleToDLPack() local
164 TensorReference tensor_ref(tensor); in SetProtoFromGPU() local
842 TensorReference tensor_ref(*t); in allocate_persistent() local