Lines Matching refs:params
28 def get_shapes(params): argument
30 …shapes["lhs_input_tensor"] = (params["lhs_batch"], params["lhs_height"], params["lhs_rows"], param…
31 …shapes["rhs_input_tensor"] = (params["rhs_batch"], params["rhs_height"], params["rhs_rows"], param…
32 …shapes["representational_dataset"] = (params["lhs_batch"], params["lhs_height"], params["lhs_rows"…
33 …shapes["representational_dataset2"] = (params["rhs_batch"], params["rhs_height"], params["rhs_rows…
38 def generate_keras_model(shapes, params): argument
40 …input_shape_lhs = (params["lhs_batch"], params["lhs_height"], params["lhs_rows"], params["lhs_cols…
41 …input_shape_rhs = (params["rhs_batch"], params["rhs_height"], params["rhs_rows"], params["rhs_cols…
45 … layer = tf.matmul(input_lhs, input_rhs, transpose_a=params["adj_x"], transpose_b=params["adj_y"])
50 def generate_data_tflite(tflite_fname, params): argument
80 minval = Lib.op_utils.get_dtype_min(params["input_data_type"])
81 maxval = Lib.op_utils.get_dtype_max(params["input_data_type"])