Searched defs:tensor_ref (Results 1 – 11 of 11) sorted by relevance
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()
372 etdump_Tensor_ref_t tensor_ref = add_tensor_entry(builder_, output, offset); in log_intermediate_output_delegate_helper() local534 etdump_Tensor_ref_t tensor_ref = in log_evalue() local
89 const TensorRef tensor_ref = val.toTensorRef(); in print_readable() local
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()
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