| /external/tensorflow/tensorflow/lite/toco/ |
| D | tensorflow_util.cc | 35 namespace toco { namespace
50 toco::port::AppendF(&dump, R"MSG( in LogDumpGraphDef()
57 toco::port::AppendF(&dump, " %s\n", op); in LogDumpGraphDef()
63 toco::port::AppendF(&dump, R"MSG( in LogDumpGraphDef()
70 toco::port::AppendF(&dump, " %s\n", input); in LogDumpGraphDef()
74 toco::port::AppendF(&dump, " ATTR: name = %s\n", attr.first); in LogDumpGraphDef()
78 toco::port::AppendF(&dump, " placeholder: %s\n", in LogDumpGraphDef()
87 toco::port::AppendF(&dump, " %s\n", line); in LogDumpGraphDef()
93 toco::port::AppendF(&dump, " int: %lld\n", attr.second.i()); in LogDumpGraphDef()
95 toco::port::AppendF(&dump, " float: %g\n", attr.second.f()); in LogDumpGraphDef()
[all …]
|
| D | toco.cc | 24 toco::string msg; in main()
25 toco::ParsedTocoFlags parsed_toco_flags; in main()
26 toco::ParsedModelFlags parsed_model_flags; in main()
41 bool toco_success = toco::ParseTocoFlagsFromCommandLineFlags( in main()
43 bool model_success = toco::ParseModelFlagsFromCommandLineFlags( in main()
50 toco::port::InitGoogle(argv[0], effective_argc, &effective_argv, true); in main()
51 auto status = toco::Convert(parsed_toco_flags, parsed_model_flags); in main()
|
| D | args.h | 29 namespace toco {
82 class Arg<toco::IntList> final {
95 const toco::IntList& value() const { return parsed_value_; } in value()
98 toco::IntList parsed_value_;
103 class Arg<toco::StringMapList> final {
117 const toco::StringMapList& value() const { return parsed_value_; } in value()
120 toco::StringMapList parsed_value_;
139 Arg<toco::IntList> input_shape;
140 Arg<toco::StringMapList> rnn_states;
141 Arg<toco::StringMapList> model_checks;
|
| D | toco_tooling.cc | 35 namespace toco { namespace
143 bool IsRealValued(toco::ArrayDataType type) { in IsRealValued()
149 return static_cast<bool>(type == toco::ArrayDataType::kFloat || in IsRealValued()
150 type == toco::ArrayDataType::kUint8 || in IsRealValued()
151 type == toco::ArrayDataType::kInt16); in IsRealValued()
194 if (array->data_type != toco::ArrayDataType::kFloat && in SetFinalDataTypeOnInputs()
195 type != toco::ArrayDataType::kFloat) { in SetFinalDataTypeOnInputs()
225 model = toco::tflite::Import(model_flags, input_file_contents); in Import()
263 model, "Removing unused ops", {new toco::RemoveUnusedOp})); in TransformWithStatus()
297 transformations.Add(new toco::SplitLstmCellInputs); in TransformWithStatus()
[all …]
|
| D | BUILD | 295 # exposed by the :toco command-line tool.
334 "//tensorflow/lite/toco/tensorflow_graph_matching:resolve_cluster",
335 "//tensorflow/lite/toco/tflite:export",
336 "//tensorflow/lite/toco/tflite:import",
401 # :toco is the main public command-line tool exposing the functionality
420 # it can conflict with downstream deps when toco is used as a library.
426 name = "toco",
427 srcs = ["toco.cc"],
442 # it can conflict with downstream deps when toco is used as a library.
465 # it can conflict with downstream deps when toco is used as a library.
|
| D | toco_cmdline_flags.cc | 29 namespace toco { namespace
225 namespace port = toco::port; in ReadTocoFlagsFromCommandLineFlags()
290 toco::IODataType input_type; in ReadTocoFlagsFromCommandLineFlags()
291 QCHECK(toco::IODataType_Parse(parsed_toco_flags.input_type.value(), in ReadTocoFlagsFromCommandLineFlags()
309 toco::IODataType input_type; in ReadTocoFlagsFromCommandLineFlags()
310 QCHECK(toco::IODataType_Parse(input_types[0], &input_type)); in ReadTocoFlagsFromCommandLineFlags()
|
| D | model_cmdline_flags_test.cc | 25 namespace toco { namespace
65 ::toco::port::InitGoogleWasDoneElsewhere(); in main()
|
| D | toco_port.cc | 29 namespace toco { namespace
53 namespace toco { namespace
82 toco::port::file::Options ToOptions(const ::file::Options& options) { in ToOptions()
140 namespace toco { namespace
|
| D | toco_port_test.cc | 22 namespace toco { namespace
64 ::toco::port::InitGoogleWasDoneElsewhere(); in main()
|
| D | args.cc | 19 namespace toco { namespace
112 bool Arg<toco::IntList>::Parse(string text) { in Parse()
128 bool Arg<toco::StringMapList>::Parse(string text) { in Parse()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | BUILD | 14 "//tensorflow/lite/toco:graph_transformations",
15 "//tensorflow/lite/toco:model",
16 "//tensorflow/lite/toco:tooling_util",
25 "//tensorflow/lite/toco:graph_transformations",
26 "//tensorflow/lite/toco:model",
27 "//tensorflow/lite/toco:tooling_util",
36 "//tensorflow/lite/toco:graph_transformations",
37 "//tensorflow/lite/toco:model",
38 "//tensorflow/lite/toco:tooling_util",
48 "//tensorflow/lite/toco:graph_transformations",
[all …]
|
| D | resolve_constant_concatenation_test.cc | 24 namespace toco { namespace
134 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>(); in PrepareModel()
137 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>().data.data(); in PrepareModel()
167 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
176 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
189 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
198 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
211 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
220 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
|
| D | fuse_binary_into_following_affine_test.cc | 24 namespace toco { namespace
92 toco::FuseBinaryIntoFollowingAffine transformation; in TEST_F()
104 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
108 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
140 toco::FuseBinaryIntoFollowingAffine transformation; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 29 from tensorflow.lite.toco import model_flags_pb2 as _model_flags_pb2
30 from tensorflow.lite.toco import toco_flags_pb2 as _toco_flags_pb2
31 from tensorflow.lite.toco import types_pb2 as _types_pb2
321 toco = _toco_flags_pb2.TocoFlags()
322 toco.input_format = input_format
323 toco.output_format = output_format
324 toco.inference_type = convert_dtype_to_tflite_type(inference_type)
326 toco.inference_input_type = convert_dtype_to_tflite_type(
329 toco.inference_input_type = toco.inference_type
330 toco.drop_control_dependency = drop_control_dependency
[all …]
|
| D | BUILD | 136 "//tensorflow/lite/toco:toco_flags_proto_py",
148 "//tensorflow/lite/toco:model_flags_proto_py",
149 "//tensorflow/lite/toco:toco_flags_proto_py",
150 "//tensorflow/lite/toco/python:tensorflow_wrap_toco",
151 "//tensorflow/lite/toco/python:toco_from_protos",
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | BUILD | 29 "//tensorflow/lite/toco:model_flags_proto_cc",
30 "//tensorflow/lite/toco:toco_flags_proto_cc",
31 "//tensorflow/lite/toco:toco_graphviz_dump_options",
32 "//tensorflow/lite/toco:toco_port",
33 "//tensorflow/lite/toco:toco_tooling",
46 srcs = ["toco.i"],
54 "//tensorflow/lite/toco:model_flags_proto_cc",
55 "//tensorflow/lite/toco:toco_flags_proto_cc",
76 "//tensorflow/lite/toco:model_flags_proto_py",
77 "//tensorflow/lite/toco:toco_flags_proto_py",
|
| D | toco_python_api.cc | 28 namespace toco { namespace
60 toco::ModelFlags model_flags; in TocoConvert()
64 toco::TocoFlags toco_flags; in TocoConvert()
78 std::unique_ptr<toco::Model> model = in TocoConvert()
79 toco::Import(toco_flags, model_flags, input_contents_txt); in TocoConvert()
80 toco::Transform(toco_flags, model.get()); in TocoConvert()
|
| D | toco_from_protos_test.py | 23 from tensorflow.lite.toco import model_flags_pb2
24 from tensorflow.lite.toco import toco_flags_pb2
25 from tensorflow.lite.toco import types_pb2
|
| /external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
| D | BUILD | 19 "//tensorflow/lite/toco:toco_port",
34 "//tensorflow/lite/toco:model",
35 "//tensorflow/lite/toco:tooling_util",
53 "//tensorflow/lite/toco:model",
54 "//tensorflow/lite/toco:toco_port",
55 "//tensorflow/lite/toco:tooling_util",
88 "//tensorflow/lite/toco:tooling_util",
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | BUILD | 32 "//tensorflow/lite/toco:graph_transformations",
33 "//tensorflow/lite/toco:model",
48 "//tensorflow/lite/toco:tooling_util",
65 "//tensorflow/lite/toco:model",
95 "//tensorflow/lite/toco:model",
96 "//tensorflow/lite/toco:tooling_util",
130 "//tensorflow/lite/toco:model",
131 "//tensorflow/lite/toco:tooling_util",
|
| D | export_test.cc | 25 namespace toco { namespace
510 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
532 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
546 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
561 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
574 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
590 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
608 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
633 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
649 const toco::OperatorSignature op_signature = {op.get(), &model}; in TEST()
|
| D | export.cc | 27 namespace toco { namespace
109 const ::toco::OperatorSignature& op_signature, in OperatorKey()
113 const ::toco::Operator& op = *op_signature.op; in OperatorKey()
194 const toco::OperatorSignature op_signature = {op.get(), &model}; in LoadOperatorsMap()
217 const toco::Array& array = *array_pair.second; in ExportTensors()
307 const toco::OperatorSignature op_signature = {op.get(), &model}; in ExportOperatorCodes()
356 const toco::OperatorSignature op_signature = {op.get(), &model}; in ExportOperators()
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 164 "$(location //tensorflow/lite/toco:toco)",
175 tools = ["//tensorflow/lite/toco:toco"] + tf_binary_additional_srcs(),
341 if conversion_mode == "toco-flex":
355 return ["toco-flex", ""]
380 if conversion_mode == "toco-flex":
395 toco = "//tensorflow/lite/toco:toco"
400 # toco = "//tensorflow/lite/experimental/pb2lite:pb2lite"
406 toco = toco,
411 def gen_zipped_test_file(name, file, toco, flags):
417 toco: str. Pathname of toco binary to run
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | graph_transformations.cc | 28 namespace toco { namespace
196 PrintModelStats(toco::port::StringF("Before %s", msg), *model); in RunGraphTransformationsWithStatus()
203 toco::port::StringF("After %s pass %d", msg, pass_index); in RunGraphTransformationsWithStatus()
|
| D | ensure_uint8_weights_safe_for_fast_int8_kernels.cc | 25 namespace toco { namespace
125 const auto& fc_op = static_cast<const toco::FullyConnectedOperator&>(op); in Run()
|