Searched refs:variable_tensor (Results 1 – 3 of 3) sorted by relevance
69 TfLiteTensor* variable_tensor = variable->GetTensor(); in Eval() local74 TF_LITE_ENSURE_TYPES_EQ(context, variable_tensor->type, output->type); in Eval()79 TfLiteIntArrayCopy(variable_tensor->dims))); in Eval()81 memcpy(output->data.raw, variable_tensor->data.raw, output->bytes); in Eval()
335 for (const auto& [variable_index, variable_tensor] : variables_snapshot) { in ComputeAsync()336 snapshot_ptrs.emplace(variable_index, variable_tensor.has_value() in ComputeAsync()337 ? &variable_tensor.value() in ComputeAsync()602 for (const auto& [variable_index, variable_tensor] : in Compute()604 snapshot_ptrs.emplace(variable_index, variable_tensor.has_value() in Compute()605 ? &variable_tensor.value() in Compute()
2043 const Tensor* variable_tensor = variable->tensor(); in InferShapesWithResourceVar() local2046 variable_tensor_vec.reserve(variable_tensor->dims()); in InferShapesWithResourceVar()2047 for (int d = 0; d < variable_tensor->dims(); ++d) { in InferShapesWithResourceVar()2048 variable_tensor_vec.push_back(variable_tensor->dim_size(d)); in InferShapesWithResourceVar()