| 782 | } |
| 783 | |
| 784 | void GraphTransferer::RegisterInputNode( |
| 785 | const IRemoteFusedGraphOpsDefinitions& ops_definitions, |
| 786 | const ShapeRefiner& shape_refiner, const Node& node) { |
| 787 | const string op_type = node.type_string(); |
| 788 | VLOG(1) << "Register input node: " << node.name() << ", " << op_type; |
| 789 | CHECK_EQ(node_name_to_id_cache_map_.count(node.name()), 1); |
| 790 | const int id = node_name_to_id_cache_map_[node.name()]; |
| 791 | // TODO(satok): Set correct data type if it's given. |
| 792 | const int op_type_id = ops_definitions.GetOpIdFor("INPUT", {}); |
| 793 | CHECK(op_type_id >= 0 && op_type_id < ops_definitions.GetTotalOpsCount()) |
| 794 | << "Op" << node.name() << ", " << op_type << " is not supported," |
| 795 | << op_type_id; |
| 796 | AppendNodeParamsWithIoParams( |
| 797 | shape_refiner, node, node.name(), id, node.type_string(), op_type_id, |
| 798 | PADDING_NA_ID, node.num_inputs(), {}, node.num_outputs(), |
| 799 | true /* append_input */, true /* append_output */); |
| 800 | } |
| 801 | |
| 802 | void GraphTransferer::RegisterFlattenNode( |
| 803 | const IRemoteFusedGraphOpsDefinitions& ops_definitions, |
nothing calls this directly
no test coverage detected