| 83 | } |
| 84 | |
| 85 | void GroupServerProxy::set_output_shape( |
| 86 | void* input_ptr, size_t input_len, std::string* output) { |
| 87 | INFO_INIT(mm_handler, SetOutputShape); |
| 88 | auto&& shape_proto = req.shape(); |
| 89 | TensorShape shape; |
| 90 | shape.ndim = shape_proto.ndim(); |
| 91 | for (size_t i = 0; i < shape.ndim; ++i) { |
| 92 | shape.shape[i] = shape_proto.shape(i); |
| 93 | } |
| 94 | m_mgr.set_output_shape(req.key(), shape); |
| 95 | rsp.SerializeToString(output); |
| 96 | } |
| 97 | |
| 98 | void GroupServerProxy::get_output_shape( |
| 99 | void* input_ptr, size_t input_len, std::string* output) { |
no test coverage detected