MCPcopy Create free account
hub / github.com/Tencent/PocketFlow / build_graph

Function build_graph

learners/uniform_quantization_tf/utils.py:151–224  ·  view source on GitHub ↗

Build a graph for training or evaluation. Args: * model_helper: model helper with definitions of model & dataset * unquant_node_names: list of unquantized activation node names * config: graph configuration * is_train: insert training-related operations or not Returns: * model: dicti

(model_helper, unquant_node_names, config, is_train)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

find_unquant_act_nodesFunction · 0.85

Calls 3

insert_quant_opFunction · 0.85
create_sessionFunction · 0.85
forward_fnFunction · 0.50

Tested by

no test coverage detected