1 licenses(["notice"]) # Apache 2.0 2 3 package(default_visibility = ["//tensorflow:internal"]) 4 5 load("//tensorflow:tensorflow.bzl", "py_test") 6 load("//tensorflow/lite:build_def.bzl", "tf_to_tflite") 7 8 exports_files(glob(["*.pb"])) 9 10 tf_to_tflite( 11 name = "permute_float", 12 src = "permute.pbtxt", 13 out = "permute_float.tflite", 14 options = [ 15 "--input_arrays=input", 16 "--output_arrays=output", 17 ], 18 ) 19 20 tf_to_tflite( 21 name = "permute_uint8", 22 src = "permute.pbtxt", 23 out = "permute_uint8.tflite", 24 options = [ 25 "--input_arrays=input", 26 "--output_arrays=output", 27 "--inference_type=QUANTIZED_UINT8", 28 "--std_values=1", 29 "--mean_values=0", 30 "--default_ranges_min=0", 31 "--default_ranges_max=255", 32 ], 33 ) 34 35 tf_to_tflite( 36 name = "gather_string", 37 src = "gather.pbtxt", 38 out = "gather_string.tflite", 39 options = [ 40 "--input_arrays=input,indices", 41 "--output_arrays=output", 42 ], 43 ) 44 45 filegroup( 46 name = "interpreter_test_data", 47 srcs = [ 48 ":gather_string", 49 ":permute_float", 50 ":permute_uint8", 51 ], 52 visibility = ["//tensorflow:__subpackages__"], 53 ) 54
