Lines Matching refs:tensor
52 int index, const TfLiteTensor** tensor);
83 int index, TfLiteTensor** tensor);
120 TfLiteTensor** tensor);
144 TfLiteTensor** tensor);
178 inline bool IsConstantTensor(const TfLiteTensor* tensor) { in IsConstantTensor() argument
179 return tensor->allocation_type == kTfLiteMmapRo; in IsConstantTensor()
184 inline bool IsDynamicTensor(const TfLiteTensor* tensor) { in IsDynamicTensor() argument
185 return tensor->allocation_type == kTfLiteDynamic; in IsDynamicTensor()
189 inline void SetTensorToDynamic(TfLiteTensor* tensor) { in SetTensorToDynamic() argument
190 if (tensor->allocation_type != kTfLiteDynamic) { in SetTensorToDynamic()
191 tensor->allocation_type = kTfLiteDynamic; in SetTensorToDynamic()
192 tensor->data.raw = nullptr; in SetTensorToDynamic()
197 inline void SetTensorToPersistentRo(TfLiteTensor* tensor) { in SetTensorToPersistentRo() argument
198 if (tensor->allocation_type != kTfLitePersistentRo) { in SetTensorToPersistentRo()
199 tensor->allocation_type = kTfLitePersistentRo; in SetTensorToPersistentRo()
200 tensor->data.raw = nullptr; in SetTensorToPersistentRo()