Home
last modified time | relevance | path

Searched refs:tensor_buffers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc253 const flatbuffers::Vector<flatbuffers::Offset<Buffer>>* tensor_buffers) { in GetLoggableTensorIndices() argument
262 (tensor_buffers->Get(buffer_index) == nullptr) || in GetLoggableTensorIndices()
263 (tensor_buffers->Get(buffer_index)->data() == nullptr) || in GetLoggableTensorIndices()
264 (tensor_buffers->Get(buffer_index)->data()->size() == 0); in GetLoggableTensorIndices()
337 auto tensor_buffers = tflite_model->buffers(); in BuildLoggingInterpreter() local
375 GetLoggableTensorIndices(op_info.inputs, tensors, tensor_buffers); in BuildLoggingInterpreter()
377 GetLoggableTensorIndices(op_info.outputs, tensors, tensor_buffers); in BuildLoggingInterpreter()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dsnapshot_util.cc220 std::vector<const TensorBuffer*> tensor_buffers; in WriteTensors() local
221 tensor_buffers.reserve(num_simple_); in WriteTensors()
234 tensor_buffers.push_back(tensor_buffer); in WriteTensors()
253 memcpy(position, tensor_buffers[buffer_index]->data(), in WriteTensors()