Searched defs:tensor_ref (Results 1 – 9 of 9) sorted by relevance
37 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
35 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
38 absl::Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref, in CreatePHWC4BufferFromTensorRef()
50 TensorRef<BHWC> tensor_ref; in TEST() local66 TensorRef<BHWC> tensor_ref; in TEST() local83 TensorRef<BHWC> tensor_ref; in TEST() local101 TensorRef<BHWC> tensor_ref; in TEST() local115 TensorRef<BHWC> tensor_ref; in TEST() local128 TensorRef<BHWC> tensor_ref; in TEST() local
159 TensorRef<BHWC>* tensor_ref) { in ConvertTfLiteTensorToTensorRef()
289 TensorReference tensor_ref(*tensor); // This will call buf_->Ref() in TFE_HandleToDLPack() local
164 TensorReference tensor_ref(tensor); in SetProtoFromGPU() local
148 for (auto tensor_ref : scratch_tensor_refs_) { in ~GpuSolver() local
193 for (const auto& tensor_ref : scratch_tensor_refs_) { in ~GpuSolver() local