Home
last modified time | relevance | path

Searched refs:schema_fb (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/
Dtest_utils.py25 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
34 schema_fb.BufferStart(builder)
35 buffer0_offset = schema_fb.BufferEnd(builder)
37 schema_fb.BufferStartDataVector(builder, 10)
49 schema_fb.BufferStart(builder)
50 schema_fb.BufferAddData(builder, buffer1_data_offset)
51 buffer1_offset = schema_fb.BufferEnd(builder)
53 schema_fb.BufferStart(builder)
54 buffer2_offset = schema_fb.BufferEnd(builder)
56 schema_fb.ModelStartBuffersVector(builder, 3)
[all …]
Dflatbuffer_utils.py33 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
41 model_object = schema_fb.Model.GetRootAsModel(model_bytearray, 0)
42 return schema_fb.ModelT.InitFromObj(model_object)
Dvisualize.py33 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
204 for name, value in schema_fb.TensorType.__dict__.items():
212 for name, value in schema_fb.BuiltinOperator.__dict__.items():
417 model_obj = schema_fb.Model.GetRootAsModel(buffer_data, 0)
418 model = schema_fb.ModelT.InitFromObj(model_obj)
/external/tensorflow/tensorflow/lite/python/
Dtest_util.py21 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
28 model = schema_fb.Model.GetRootAsModel(model_data, 0)
37 if builtin_code == schema_fb.BuiltinOperator.CUSTOM:
47 model = schema_fb.Model.GetRootAsModel(model_data, 0)
Dutil.py34 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
583 model_object = schema_fb.Model.GetRootAsModel(model_bytearray, 0)
584 model_object = schema_fb.ModelT.InitFromObj(model_object)
656 if builtin_code == schema_fb.BuiltinOperator.QUANTIZE:
715 uint8_quantization = schema_fb.QuantizationParametersT()
719 tensors[op.inputs[0]].type = schema_fb.TensorType.UINT8
749 if builtin_code == schema_fb.BuiltinOperator.DEQUANTIZE:
810 if builtin_code == schema_fb.BuiltinOperator.QUANTIZE:
815 quant_op = schema_fb.OperatorCodeT()
816 quant_op.builtinCode = schema_fb.BuiltinOperator.QUANTIZE
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/python/
Dmodify_model_interface_lib.py22 from tensorflow.lite.python import schema_py_generated as schema_fb unknown
47 dtype_int = schema_fb.TensorType.__dict__[dtype_str]