Lines Matching refs:tensors

104 …convert_json_to_tflite(json_template_fpath, json_output_fpath, data.tensors, replacements, args.sc…
129             if shape_name in data.tensors:
130                 input_tensors[shape_name] = data.tensors[shape_name]
133                 data.tensors[shape_name] = input_tensors[shape_name].numpy()
139         data.tensors["output"] = invoke_tflite(fpaths["tflite"], input_tensors)
141         data.tensors["output"] = invoke_tflite_runtime(fpaths["tflite"], input_tensors)
144 …data.tensors["output"] = invoke_tflite_micro(fpaths["tflite"], input_tensors, params["arena_size"])
146             data.tensors["output"] = invoke_tflite_micro(fpaths["tflite"], input_tensors)
151         data.tensors["output"] = np.maximum(data.tensors["output"], params["activation_min"])
153         data.tensors["output"] = np.minimum(data.tensors["output"], params["activation_max"])
169     for name, tensor in data.tensors.items():
387 def convert_json_to_tflite(json_template_fpath, json_output_fpath, tensors, params, schema_path):  argument
412                 for key in tensors:
420                         for weight in tensors[key].flatten():