HomeSort by relevance Sort by last modified time
    Searched refs:tensor_buffer (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/
string_util.cc 78 // Output tensor will take over the ownership of tensor_buffer, and free it
80 char* tensor_buffer = static_cast<char*>(malloc(bytes)); local
83 memcpy(tensor_buffer, &num_strings, sizeof(int32_t));
89 memcpy(tensor_buffer + sizeof(int32_t) * (i + 1), &offset, sizeof(int32_t));
93 memcpy(tensor_buffer + start, data_.data(), data_.size());
95 // Set tensor content pointer to tensor_buffer, and release original data.
99 tensor_buffer, bytes, kTfLiteDynamic, tensor->allocation,
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.cc 71 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
81 CopyArrayData(sub_buffer, dim2_copy_size, 0, 0, &tensor_buffer,
lstm_utils.h 94 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,

Completed in 174 milliseconds