| 818 | } |
| 819 | |
| 820 | void GraphTransferer::RegisterGenericNode( |
| 821 | const IRemoteFusedGraphOpsDefinitions& ops_definitions, |
| 822 | const ShapeRefiner& shape_refiner, const Node& node) { |
| 823 | VLOG(1) << "Register generic node: " << node.name(); |
| 824 | CHECK_EQ(node_name_to_id_cache_map_.count(node.name()), 1); |
| 825 | const int id = node_name_to_id_cache_map_[node.name()]; |
| 826 | // TODO(satok): Set correct data type if it's given. |
| 827 | const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); |
| 828 | CHECK(op_type_id >= 0 && op_type_id < ops_definitions.GetTotalOpsCount()); |
| 829 | |
| 830 | AppendNodeParamsWithIoParams( |
| 831 | shape_refiner, node, node.name(), id, node.type_string(), op_type_id, |
| 832 | PADDING_NA_ID, node.num_inputs(), {}, node.num_outputs(), |
| 833 | true /* append_input */, true /* append_output */); |
| 834 | } |
| 835 | |
| 836 | // TODO(satok): Remove this function. |
| 837 | // TODO(satok): Remove only_register_const_node. |
nothing calls this directly
no test coverage detected