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

Function to_shapes

src/argument.cpp:125–132  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

123}
124
125std::vector<shape> to_shapes(const std::vector<argument>& args)
126{
127 std::vector<shape> shapes;
128 std::transform(args.begin(), args.end(), std::back_inserter(shapes), [](auto&& arg) {
129 return arg.get_shape();
130 });
131 return shapes;
132}
133
134argument::argument(const std::vector<argument>& args)
135 : m_shape(to_shapes(args)), m_data(data_t::from_args(args))

Callers 15

perf_groupFunction · 0.70
insert_common_argsFunction · 0.70
get_graph_colorFunction · 0.70
try_compute_shapeFunction · 0.70
argumentMethod · 0.70
create_program_from_mlirFunction · 0.50
unroll_pointwiseFunction · 0.50
compileFunction · 0.50
applyMethod · 0.50
parseMethod · 0.50
compute_dump_nameFunction · 0.50
insert_mlirFunction · 0.50

Calls 4

transformFunction · 0.50
beginMethod · 0.45
endMethod · 0.45
get_shapeMethod · 0.45

Tested by

no test coverage detected