| 16 | } |
| 17 | |
| 18 | bool CustomKernelNodeOps::OnBind(Node* node) |
| 19 | { |
| 20 | if(!k_ops_->inplace_info) |
| 21 | return true; |
| 22 | |
| 23 | int output_num = node->GetOutputNum(); |
| 24 | |
| 25 | inplace_t io_map; |
| 26 | |
| 27 | for(int i = 0; i < output_num; i++) |
| 28 | { |
| 29 | int in_idx = k_ops_->inplace_info(k_ops_, i); |
| 30 | |
| 31 | if(in_idx < 0) |
| 32 | continue; |
| 33 | |
| 34 | io_map[i] = in_idx; |
| 35 | } |
| 36 | |
| 37 | return true; |
| 38 | } |
| 39 | |
| 40 | bool CustomKernelNodeOps::Prerun(Node* node) |
| 41 | { |
no test coverage detected