Function GetVariableInput

tensorflow/lite/kernels/kernel_util.h:40–48

  return &context->tensors[tensor_index];
}
inline TfLiteTensor* GetVariableInput(TfLiteContext* context, TfLiteNode* node,
                                      int index) {
  const int tensor_index = flatbuffers::EndianScalar(node->inputs->data[index]);
  if (tensor_index < 0) {
    return nullptr;
  }
  TfLiteTensor* tensor = &context->tensors[tensor_index];
  return (tensor->is_variable) ? tensor : nullptr;
}
inline TfLiteTensor* GetOutput(TfLiteContext* context, TfLiteNode* node,
                               int index) {
  const int tensor_index = flatbuffers::EndianScalar(node->outputs->data[index]);