pymoose.predictors.predictor_utils module#

pymoose.predictors.predictor_utils.find_activation_in_model_proto(model_proto, operator_name, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_attribute_in_node(node, attribute_name, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_initializer_in_model_proto(model_proto, operator_name, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_input_shape(input_node)[source]#
pymoose.predictors.predictor_utils.find_node_in_model_proto(model_proto, operator_name, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_op_types_in_model_proto(model_proto, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_output_in_model_proto(model_proto, enforce=True)[source]#
pymoose.predictors.predictor_utils.find_parameters_in_model_proto(model_proto, operator_names, enforce=True)[source]#