Searched refs:const_node_info (Results 1 – 5 of 5) sorted by relevance
483 GraphTransferConstNodeInfo& const_node_info = in RegisterConstantNode() local485 const_node_info.set_name(node.name()); in RegisterConstantNode()486 const_node_info.set_node_id(id); in RegisterConstantNode()489 const_node_info.add_shape(shape_array[0]); in RegisterConstantNode()490 const_node_info.add_shape(shape_array[1]); in RegisterConstantNode()491 const_node_info.add_shape(shape_array[2]); in RegisterConstantNode()492 const_node_info.add_shape(shape_array[3]); in RegisterConstantNode()498 const_node_info.set_dtype(const_tensor.dtype()); in RegisterConstantNode()500 const_node_info.set_data(const_tensor.tensor_data().data(), data_size); in RegisterConstantNode()515 GraphTransferConstNodeInfo& const_node_info = in RegisterConstantShape() local[all …]
339 const GraphTransferConstNodeInfo& cni0 = gfi0.const_node_info(i); in CompareGraphTransferInfo()340 const GraphTransferConstNodeInfo& cni1 = gfi1.const_node_info(i); in CompareGraphTransferInfo()
248 graph_transfer_info.const_node_info()) { in SetupGraph()
197 gt.GetGraphTransferInfo().const_node_info()) { in FindConstNodeInfo()
61 repeated GraphTransferConstNodeInfo const_node_info = 2; field