Method
remove_mem_block_by_name
(self, mem_list, tensor_name)
Source from the content-addressed store, hash-verified
| 73 | return mem_size * data_type_bytes |
| 74 | |
| 75 | def remove_mem_block_by_name(self, mem_list, tensor_name): |
| 76 | return_mem_block = None |
| 77 | for mem_block in mem_list: |
| 78 | if tensor_name == mem_block.tensor_name: |
| 79 | return_mem_block = mem_block |
| 80 | mem_list.remove(mem_block) |
| 81 | break |
| 82 | return return_mem_block |
| 83 | |
| 84 | def fake_new(self, op): |
| 85 | output_size = len(op.output) |
Tested by
no test coverage detected