Searched refs:tensorflow_graph (Results 1 – 4 of 4) sorted by relevance
149 const GraphDef& tensorflow_graph) { in HasAlreadyExportedConst() argument150 for (const auto& node : tensorflow_graph.node()) { in HasAlreadyExportedConst()162 GraphDef* tensorflow_graph, in ConvertFloatTensorConst() argument164 if (HasAlreadyExportedConst(name, *tensorflow_graph)) { in ConvertFloatTensorConst()167 tensorflow::NodeDef* const_op = tensorflow_graph->add_node(); in ConvertFloatTensorConst()180 GraphDef* tensorflow_graph) { in ConvertFloatTensorConst() argument182 output_axes_order, tensorflow_graph, in ConvertFloatTensorConst()189 GraphDef* tensorflow_graph) { in ConvertFloatTensorConst() argument190 if (HasAlreadyExportedConst(name, *tensorflow_graph)) { in ConvertFloatTensorConst()201 output_axes_order, tensorflow_graph); in ConvertFloatTensorConst()[all …]
53 explicit StatSummarizer(const tensorflow::GraphDef& tensorflow_graph);
39 StatSummarizer::StatSummarizer(const tensorflow::GraphDef& tensorflow_graph) in StatSummarizer() argument
268 tensorflow::GraphDef tensorflow_graph; in InitializeSession() local