Searched refs:shape_sig_to_inputs (Results 1 – 1 of 1) sorted by relevance
552 std::unordered_map<string, std::vector<InputAndShape>> shape_sig_to_inputs; in RewriteOptimizedNodesGroup() local554 shape_sig_to_inputs[ShapeSignature(input.shape)].push_back(input); in RewriteOptimizedNodesGroup()557 using SigKV = decltype(shape_sig_to_inputs)::value_type; in RewriteOptimizedNodesGroup()558 VLOG(3) << "Add/AddN group has " << shape_sig_to_inputs.size() in RewriteOptimizedNodesGroup()560 << absl::StrJoin(shape_sig_to_inputs, ", ", in RewriteOptimizedNodesGroup()567 shapes.reserve(shape_sig_to_inputs.size()); in RewriteOptimizedNodesGroup()568 for (const auto& el : shape_sig_to_inputs) in RewriteOptimizedNodesGroup()604 const auto& inputs = shape_sig_to_inputs[ShapeSignature(shapes[i])]; in RewriteOptimizedNodesGroup()