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

Function visit_flatten_args

src/targets/gpu/code_object_op.cpp:58–64  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

56
57template <class F>
58static void visit_flatten_args(const std::vector<argument>& args, F f)
59{
60 if(needs_flatten(args))
61 f(flatten(args));
62 else
63 f(args);
64}
65
66argument
67code_object_op::compute(context& ctx, const shape&, const std::vector<argument>& args) const

Callers 1

computeMethod · 0.85

Calls 2

needs_flattenFunction · 0.85
flattenClass · 0.50

Tested by

no test coverage detected