MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get_output_shape

Method get_output_shape

src/opr-mm/impl/mm_handler.cpp:98–108  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

96}
97
98void GroupServerProxy::get_output_shape(
99 void* input_ptr, size_t input_len, std::string* output) {
100 INFO_INIT(mm_handler, GetOutputShape);
101 auto shape = m_mgr.get_output_shape(req.key());
102 auto&& shape_proto = *rsp.mutable_shape();
103 shape_proto.set_ndim(shape.ndim);
104 for (size_t i = 0; i < shape.ndim; ++i) {
105 shape_proto.add_shape(shape[i]);
106 }
107 rsp.SerializeToString(output);
108}
109
110void GroupServerProxy::bcast_addr(
111 void* input_ptr, size_t input_len, std::string* output) {

Callers 1

Calls 3

keyMethod · 0.45
ndimMethod · 0.45
shapeMethod · 0.45

Tested by

no test coverage detected