Home
last modified time | relevance | path

Searched refs:flatbuffer_tensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers.cc96 TfLiteStatus BytesRequiredForTensor(const tflite::Tensor& flatbuffer_tensor, in BytesRequiredForTensor() argument
102 if (flatbuffer_tensor.shape() != nullptr) { in BytesRequiredForTensor()
103 for (size_t n = 0; n < flatbuffer_tensor.shape()->Length(); ++n) { in BytesRequiredForTensor()
104 element_count *= flatbuffer_tensor.shape()->Get(n); in BytesRequiredForTensor()
109 TF_LITE_ENSURE_STATUS(ConvertTensorType(flatbuffer_tensor.type(), in BytesRequiredForTensor()
Dmicro_allocator.cc430 const tflite::Tensor& flatbuffer_tensor, in GetFlatbufferTensorBuffer() argument
440 if (auto* buffer = (*buffers)[flatbuffer_tensor.buffer()]) { in GetFlatbufferTensorBuffer()
461 const tflite::Tensor& flatbuffer_tensor, in InitializeTfLiteTensorFromFlatbuffer() argument
469 TF_LITE_ENSURE_STATUS(ConvertTensorType(flatbuffer_tensor.type(), in InitializeTfLiteTensorFromFlatbuffer()
472 result->is_variable = flatbuffer_tensor.is_variable(); in InitializeTfLiteTensorFromFlatbuffer()
474 result->data.data = GetFlatbufferTensorBuffer(flatbuffer_tensor, buffers); in InitializeTfLiteTensorFromFlatbuffer()
491 flatbuffer_tensor, &result->bytes, &type_size, error_reporter)); in InitializeTfLiteTensorFromFlatbuffer()
493 if (flatbuffer_tensor.shape() == nullptr) { in InitializeTfLiteTensorFromFlatbuffer()
503 allocator, error_reporter, flatbuffer_tensor.shape(), &(result->dims))); in InitializeTfLiteTensorFromFlatbuffer()
507 const auto* src_quantization = flatbuffer_tensor.quantization(); in InitializeTfLiteTensorFromFlatbuffer()
[all …]
Dmemory_helpers.h40 TfLiteStatus BytesRequiredForTensor(const tflite::Tensor& flatbuffer_tensor,
Dmicro_allocator.h39 const tflite::Tensor& flatbuffer_tensor,