Lines Matching refs:params
28 def get_shapes(params): argument
30 …shapes["input_tensor"] = (params["batch_size"], params["input_h"], params["input_w"], params["inpu…
35 def generate_keras_model(shapes, params): argument
38 if params["op_type"] == 'avgpool':
40 keras.layers.AveragePooling2D(pool_size=(params["filter_h"], params["filter_w"]),
41 strides=(params["stride_h"], params["stride_w"]),
42 padding=params["pad"],
44 elif params["op_type"] == 'maxpool':
46 keras.layers.MaxPooling2D(pool_size=(params["filter_h"], params["filter_w"]),
47 strides=(params["stride_h"], params["stride_w"]),
48 padding=params["pad"],
56 def generate_data_tflite(tflite_fname, params): argument
70 if params["pad"] == "SAME":
71 …ng_width = max((generated_params["output_w"] - 1) * params["stride_w"] + params["filter_w"] - para…
72 …g_height = max((generated_params["output_h"] - 1) * params["stride_h"] + params["filter_h"] - para…