MCPcopy Create free account
hub / github.com/ROCm/AMDMIGraphX / assign_op

Function assign_op

src/split_reduce.cpp:150–160  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

148} // namespace
149
150static std::string assign_op(const std::vector<instruction_ref>& splits)
151{
152 static std::unordered_map<std::string, std::string> m = {
153 {"reduce_sum", "assign_add"},
154 {"reduce_mean", "assign_add"},
155 {"reduce_prod", "assign_mul"},
156 {"reduce_max", "assign_max"},
157 {"reduce_min", "assign_min"},
158 };
159 return m.at(splits.front()->name());
160}
161
162static std::vector<instruction_ref>
163insert_module_inline(module& m, instruction_ref ins, const module::with_inputs& mwi)

Callers 1

applyMethod · 0.85

Calls 3

atMethod · 0.80
frontMethod · 0.80
nameMethod · 0.45

Tested by

no test coverage detected