Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc337 auto& linkable_node = nodes_[next_nodes[0]]; in Merge() local
338 if (!linkable_node.task.IsLinkable() || linkable_node.outputs.size() != 1 || in Merge()
339 !IsReady(ready_tensors, linkable_node)) { in Merge()
344 linkable_node.task.GetDefinition().dst_tensors[0]; in Merge()
348 RETURN_IF_ERROR(MergeNodes(&linkable_node, &node)); in Merge()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc430 auto& linkable_node = nodes_[next_nodes[0]]; in Merge() local
431 if (!linkable_node.cl_operation.GetGpuOperation().IsLinkable() || in Merge()
432 linkable_node.outputs.size() != 1 || in Merge()
433 !IsReady(ready_tensors, linkable_node)) { in Merge()
439 linkable_node.cl_operation.GetDefinition().dst_tensors[0]; in Merge()
443 RETURN_IF_ERROR(MergeCLNodes(&linkable_node, &node)); in Merge()