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():