Searched refs:tensor (Results 1 – 1 of 1) sorted by relevance
123 TfLiteTensor* tensor = runner->input_tensor(name); in findInputTensor() local124 LOG_ALWAYS_FATAL_IF(!tensor, "Failed to find input tensor '%s'", name); in findInputTensor()125 return tensor; in findInputTensor()130 const TfLiteTensor* tensor = runner->output_tensor(name); in findOutputTensor() local131 LOG_ALWAYS_FATAL_IF(!tensor, "Failed to find output tensor '%s'", name); in findOutputTensor()132 return tensor; in findOutputTensor()138 TfLiteTensor*>::type tensor) { in getTensorBuffer() argument139 LOG_ALWAYS_FATAL_IF(!tensor); in getTensorBuffer()142 LOG_ALWAYS_FATAL_IF(tensor->type != type, "Unexpected type for '%s' tensor: %s (expected %s)", in getTensorBuffer()143 tensor->name, TfLiteTypeGetName(tensor->type), TfLiteTypeGetName(type)); in getTensorBuffer()[all …]