Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.cc617 std::map<ValueId, ValueId> remap_from_graph_ids; in AllocateMemoryForStrongShapes() local
619 remap_from_graph_ids[usage.first] = usage_records.size(); in AllocateMemoryForStrongShapes()
637 const auto id = assignment.object_ids[remap_from_graph_ids[tensor_id]]; in AllocateMemoryForStrongShapes()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.cc626 std::map<ValueId, ValueId> remap_from_graph_ids; in AllocateStrongShapesTensors() local
628 remap_from_graph_ids[usage.first] = usage_records.size(); in AllocateStrongShapesTensors()
651 const auto id = assignment.object_ids[remap_from_graph_ids[tensor_id]]; in AllocateStrongShapesTensors()