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

Method get_parameter_shapes

src/module.cpp:698–710  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

696}
697
698std::unordered_map<std::string, shape> module::get_parameter_shapes() const
699{
700 std::unordered_map<std::string, shape> result;
701 for(auto&& ins : impl->instructions)
702 {
703 if(ins.name() == "@param")
704 {
705 auto&& name = any_cast<builtin::param>(ins.get_operator()).parameter;
706 result[name] = ins.get_shape();
707 }
708 }
709 return result;
710}
711
712bool module::has_instruction(instruction_ref ins) const { return impl->contains(ins); }
713

Callers 15

tune_input_shapeFunction · 0.45
mainFunction · 0.45
TEST_CASEFunction · 0.45
TEST_CASEFunction · 0.45
TEST_CASEFunction · 0.45
run_progFunction · 0.45
TEST_CASEFunction · 0.45
generate_paramsFunction · 0.45
run_gpuFunction · 0.45
run_targetMethod · 0.45
verifyMethod · 0.45
test_conv_reluFunction · 0.45

Calls 2

nameMethod · 0.45
get_shapeMethod · 0.45

Tested by 15

tune_input_shapeFunction · 0.36
TEST_CASEFunction · 0.36
TEST_CASEFunction · 0.36
test_conv_reluFunction · 0.36
test_conv_reluFunction · 0.36
test_sub_uint64Function · 0.36
test_neg_int64Function · 0.36
test_nonzeroFunction · 0.36
test_fp16_imagescalerFunction · 0.36
test_if_plFunction · 0.36
run_progFunction · 0.36
test_conv_reluFunction · 0.36