Lines Matching refs:output_tensor
212 auto* output_tensor = mTfliteInterpreter->tensor(output); in saveInferenceOutput() local
214 sink.insert(sink.end(), output_tensor->data.uint8, in saveInferenceOutput()
215 output_tensor->data.uint8 + output_tensor->bytes); in saveInferenceOutput()
221 auto* output_tensor = mTfliteInterpreter->tensor(output); in getOutputError() local
222 if (output_tensor->bytes != length) { in getOutputError()
224 output_tensor->bytes, length); in getOutputError()
230 switch (output_tensor->type) { in getOutputError()
233 elements_count = output_tensor->bytes; in getOutputError()
234 for (size_t i = 0; i < output_tensor->bytes; ++i) { in getOutputError()
244 elements_count = output_tensor->bytes / sizeof(float); in getOutputError()
245 for (size_t i = 0; i < output_tensor->bytes / sizeof(float); ++i) { in getOutputError()
253 FATAL("Output sensor type %d not supported", output_tensor->type); in getOutputError()
639 auto* output_tensor = mTfliteInterpreter->tensor(tensor_index); in dumpAllLayers() local
640 if (output_tensor->data.raw == nullptr) { in dumpAllLayers()
649 fwrite(output_tensor->data.raw, output_tensor->bytes, 1, f); in dumpAllLayers()