Searched refs:variable_tensor (Results 1 – 2 of 2) sorted by relevance
63 TfLiteTensor* variable_tensor = variable->GetTensor(); in Eval() local68 TF_LITE_ENSURE_TYPES_EQ(context, variable_tensor->type, output->type); in Eval()71 context, output, TfLiteIntArrayCopy(variable_tensor->dims))); in Eval()72 memcpy(output->data.raw, variable_tensor->data.raw, output->bytes); in Eval()
1958 const Tensor* variable_tensor = variable->tensor(); in InferShapesWithResourceVar() local1961 variable_tensor_vec.reserve(variable_tensor->dims()); in InferShapesWithResourceVar()1962 for (int d = 0; d < variable_tensor->dims(); ++d) { in InferShapesWithResourceVar()1963 variable_tensor_vec.push_back(variable_tensor->dim_size(d)); in InferShapesWithResourceVar()