Searched refs:inference_type (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/python/
H A Dlite.py144 inference_type=FLOAT,
159 inference_type: Currently must be `{FLOAT, QUANTIZED_UINT8}`.
164 std deviation of training data. Only needed if `inference_type` is
183 toco.inference_type = inference_type
198 if inference_type == QUANTIZED_UINT8:
H A Dlite_test.py55 inference_type=lite.QUANTIZED_UINT8,
/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dtoco_cmdline_flags.cc60 Flag("inference_type", parsed_flags.inference_type.bind(),
61 parsed_flags.inference_type.default_value(),
68 "If not specified, inference_type is used. "
167 PARSE_TOCO_FLAG(IODataType, inference_type, FlagRequirement::kNone);
H A Dtoco_tooling.cc134 type = ConvertIODataTypeToArrayDataType(toco_flags.inference_type());
198 const IODataType inference_type = toco_flags.inference_type(); local
201 SupportsQuantization(output_format) && inference_type == QUANTIZED_UINT8;
H A Dargs.h224 Arg<string> inference_type; member in struct:toco::ParsedTocoFlags
/external/tensorflow/tensorflow/contrib/lite/toco/python/
H A Dtoco_from_protos_test.py52 toco_flags.inference_type = types_pb2.FLOAT
/external/tensorflow/tensorflow/contrib/lite/testing/
H A Dgenerate_examples.py129 inference_type = "FLOAT"
133 inference_type = "QUANTIZED_UINT8"
135 " --inference_type=%s" % inference_type +

Completed in 215 milliseconds