Method Profile

tensorflow/lite/delegates/gpu/cl/inference_context.cc:372–382

}

Status InferenceContext::Profile(ProfilingCommandQueue* queue,
                                 ProfilingInfo* result) {
  queue->ResetMeasurements();
  for (auto& node : nodes_) {
    queue->SetEventsLabel(node.name);
    RETURN_IF_ERROR(node.operations[0]->AddToQueue(queue));
  }
  RETURN_IF_ERROR(queue->WaitForCompletion());
  *result = queue->GetProfilingInfo();
  return OkStatus();
}

Tensor* InferenceContext::GetTensor(ValueId id) {
  return &tensors_[remap_from_graph_ids_to_shared_[id]];