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

Method get_parameter_shape

src/module.cpp:618–635  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

616}
617
618shape module::get_parameter_shape(std::string name) const
619{
620 auto ins = std::find_if(
621 impl->instructions.begin(), impl->instructions.end(), [&](const instruction& x) {
622 if(x.name() == "@param")
623 {
624 return any_cast<builtin::param>(x.get_operator()).parameter == name;
625 }
626 else
627 {
628 return false;
629 }
630 });
631 if(ins != this->end())
632 return ins->get_shape();
633 else
634 return {};
635}
636
637std::vector<std::string> module::get_parameter_names() const
638{

Callers 5

TEST_CASEFunction · 0.45
TEST_CASEFunction · 0.45
create_program_from_mlirFunction · 0.45
applyMethod · 0.45
run_loopFunction · 0.45

Calls 5

endMethod · 0.95
find_ifFunction · 0.50
beginMethod · 0.45
nameMethod · 0.45
get_shapeMethod · 0.45

Tested by 2

TEST_CASEFunction · 0.36
TEST_CASEFunction · 0.36