Searched refs:eval_tensor (Results 1 – 3 of 3) sorted by relevance
91 TfLiteEvalTensor* eval_tensor = in GetEvalTensor() local94 TFLITE_DCHECK(eval_tensor != nullptr); in GetEvalTensor()98 eval_tensor->data = runner->tensors_[tensor_index].data; in GetEvalTensor()99 eval_tensor->dims = runner->tensors_[tensor_index].dims; in GetEvalTensor()100 eval_tensor->type = runner->tensors_[tensor_index].type; in GetEvalTensor()101 return eval_tensor; in GetEvalTensor()
116 TfLiteStatus TfLiteEvalTensorByteLength(const TfLiteEvalTensor* eval_tensor, in TfLiteEvalTensorByteLength() argument122 if (eval_tensor->dims != nullptr) { in TfLiteEvalTensorByteLength()123 for (int n = 0; n < eval_tensor->dims->size; ++n) { in TfLiteEvalTensorByteLength()124 element_count *= eval_tensor->dims->data[n]; in TfLiteEvalTensorByteLength()128 TF_LITE_ENSURE_STATUS(TfLiteTypeSizeOf(eval_tensor->type, &type_size)); in TfLiteEvalTensorByteLength()
46 TfLiteStatus TfLiteEvalTensorByteLength(const TfLiteEvalTensor* eval_tensor,