Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc554 std::map<ValueId, ValueId> remap_from_graph_ids; in AllocateMemoryForStrongShapes() local
556 remap_from_graph_ids[usage.first] = usage_records.size(); in AllocateMemoryForStrongShapes()
574 const auto id = assignment.object_ids[remap_from_graph_ids[tensor_id]]; in AllocateMemoryForStrongShapes()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc584 std::map<ValueId, ValueId> remap_from_graph_ids; in AllocateMemoryForStrongShapes() local
586 remap_from_graph_ids[usage.first] = usage_records.size(); in AllocateMemoryForStrongShapes()
603 const auto id = assignment.object_ids[remap_from_graph_ids[t.first]]; in AllocateMemoryForStrongShapes()