Searched refs:sizes_node (Results 1 – 3 of 3) sorted by relevance
60 auto sizes_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local64 AddInput(TensorID(sizes_node->GetID(), 0)); in PopulateSubGraph()
976 Node* sizes_node = graph->insertNode( in buildShapeExpressions() local978 sizes_node->i_(attr::dim, n->i(attr::dim)); in buildShapeExpressions()979 sizes_node->i_(attr::chunks, n->i(attr::chunks)); in buildShapeExpressions()980 for (Value* output : sizes_node->outputs()) { in buildShapeExpressions()983 Value* regular_size = sizes_node->outputs().at(0); in buildShapeExpressions()984 Value* last_size = sizes_node->outputs().at(1); in buildShapeExpressions()
487 Node* sizes_node = graph->insertNode( in buildShapeExpressions() local489 sizes_node->i_(attr::dim, n->i(attr::dim)); in buildShapeExpressions()490 sizes_node->i_(attr::chunks, n->i(attr::chunks)); in buildShapeExpressions()491 for (Value* output : sizes_node->outputs()) { in buildShapeExpressions()494 Value* regular_size = sizes_node->outputs().at(0); in buildShapeExpressions()495 Value* last_size = sizes_node->outputs().at(1); in buildShapeExpressions()