Searched refs:BufferMap (Results 1 – 6 of 6) sorted by relevance
131 BufferMap::BufferMap() {} in BufferMap() function in tflite::flex::BufferMap133 BufferMap::~BufferMap() {} in ~BufferMap()135 bool BufferMap::HasTensor(int tensor_index) const { in HasTensor()139 bool BufferMap::IsTensorFlowTensor(int tensor_index) const { in IsTensorFlowTensor()143 tensorflow::Tensor BufferMap::GetTensor(int tensor_index) const { in GetTensor()147 void BufferMap::SetFromTfLite(int tensor_index, const TfLiteTensor* tensor) { in SetFromTfLite()171 void BufferMap::SetFromTensorFlow(int tensor_index, tensorflow::Tensor tensor) { in SetFromTensorFlow()
80 BufferMap buffer_map; // BufferMap is the easiest way to build the tensor. in MakeTensor()101 BufferMap buffer_map; in TEST()106 BufferMap buffer_map; in TEST()124 BufferMap buffer_map; in TEST()147 BufferMap buffer_map; in TEST()161 BufferMap buffer_map; in TEST()173 BufferMap buffer_map; in TEST()190 BufferMap buffer_map; in TEST()204 BufferMap buffer_map; in TEST()218 BufferMap buffer_map; in TEST()
32 class BufferMap {34 BufferMap();35 ~BufferMap();
45 BufferMap* GetBufferMap(const TfLiteContext* context) { in GetBufferMap()54 std::unordered_map<const TfLiteContext*, BufferMap> buffer_map_;
275 tensorflow::Status BuildEagerInputs(const BufferMap* buffer_map) { in BuildEagerInputs()302 tensorflow::Status PersistEagerOutputs(BufferMap* buffer_map) { in PersistEagerOutputs()334 tensorflow::Status ExecuteFlexOp(TfLiteContext* context, BufferMap* buffer_map, in ExecuteFlexOp()363 BufferMap* buffer_map;464 BufferMap* buffer_map = op_data->buffer_map; in Prepare()514 BufferMap* buffer_map = op_data->buffer_map; in Eval()
79 BufferMap* buffer_map = in CopyFromBufferHandle()