Searched refs:reshape_node (Results 1 – 2 of 2) sorted by relevance
763 NodeDef reshape_node; in QuantizeNodes() local764 reshape_node.set_op("Reshape"); in QuantizeNodes()765 reshape_node.set_name(unique_input_name + "/reshape"); in QuantizeNodes()766 SetNodeAttr("T", DT_FLOAT, &reshape_node); in QuantizeNodes()767 AddNodeInput(input_name, &reshape_node); in QuantizeNodes()768 AddNodeInput(reshape_dims.name(), &reshape_node); in QuantizeNodes()769 new_nodes->push_back(reshape_node); in QuantizeNodes()776 AddNodeInput(reshape_node.name(), &min_node); in QuantizeNodes()785 AddNodeInput(reshape_node.name(), &max_node); in QuantizeNodes()
1083 auto reshape_node = vectorized->node_def( in TEST() local1086 strings::StrCat(reshape_node.name(), ":output:0")); in TEST()