Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc36 inline TfLiteTensor* GetTensorAtIndex(const TfLiteContext* context, in GetTensorAtIndex() function
87 return GetTensorAtIndex(context, tensor_index); in GetMutableInput()
97 *tensor = GetTensorAtIndex(context, tensor_index); in GetMutableInputSafe()
126 return GetTensorAtIndex(context, tensor_index); in GetOutput()
135 *tensor = GetTensorAtIndex(context, tensor_index); in GetOutputSafe()
152 return GetTensorAtIndex(context, tensor_index); in GetTemporary()
162 *tensor = GetTensorAtIndex(context, tensor_index); in GetTemporarySafe()
173 return GetTensorAtIndex(context, tensor_index); in GetIntermediates()
183 *tensor = GetTensorAtIndex(context, tensor_index); in GetIntermediatesSafe()