Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc63 bool IsReady(const std::set<ValueId>& ready_tensors, const MetalNode& node) { in IsReady() argument
65 if (ready_tensors.find(in_id) == ready_tensors.end()) { in IsReady()
312 std::set<ValueId> ready_tensors; in Merge() local
314 ready_tensors.insert(input_id); in Merge()
319 ready_tensors.insert(out_id); in Merge()
339 !IsReady(ready_tensors, linkable_node)) { in Merge()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc53 bool IsReady(const absl::flat_hash_set<ValueId>& ready_tensors, in IsReady() argument
56 if (ready_tensors.find(in_id) == ready_tensors.end()) { in IsReady()
405 absl::flat_hash_set<ValueId> ready_tensors; in Merge() local
407 ready_tensors.insert(input_id); in Merge()
412 ready_tensors.insert(out_id); in Merge()
433 !IsReady(ready_tensors, linkable_node)) { in Merge()