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

Method qparam_broadcast_op

src/op/builder/convolution.cpp:300–309  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

298 }
299
300 auto qparam_broadcast_op(instruction_ref qparam,
301 std::vector<std::size_t> lens,
302 std::size_t axis) const
303 {
304 if(qparam->get_shape().elements() == 1)
305 {
306 return migraphx::make_op("multibroadcast", {{"out_lens", lens}});
307 }
308 return migraphx::make_op("broadcast", {{"out_lens", lens}, {"axis", axis}});
309 }
310
311 instruction_ref handle_quant_bias(module& m,
312 instruction_ref ins,

Callers

nothing calls this directly

Calls 3

make_opFunction · 0.50
elementsMethod · 0.45
get_shapeMethod · 0.45

Tested by

no test coverage detected