(arg_data: ArgData)
| 173 | |
| 174 | |
| 175 | def arg_data_node(arg_data: ArgData): |
| 176 | node = fn.external_source(dummy_source(arg_data.data), layout=arg_data.desc.layout) |
| 177 | if arg_data.desc.dest_device == "gpu": |
| 178 | node = node.gpu() |
| 179 | expandable_prefix = arg_data.desc.expandable_prefix |
| 180 | if expandable_prefix and expandable_prefix[0] == "F": |
| 181 | node = fn.per_frame(node) |
| 182 | return node |
| 183 | |
| 184 | |
| 185 | def as_batch(tensor): |
no test coverage detected