| /external/tensorflow/tensorflow/contrib/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 | 35 namespace toco { namespace
91 toco::string msg; in main()
92 toco::ParsedTocoFlags parsed_toco_flags; in main()
93 toco::ParsedModelFlags parsed_model_flags; in main()
108 bool toco_success = toco::ParseTocoFlagsFromCommandLineFlags( in main()
110 bool model_success = toco::ParseModelFlagsFromCommandLineFlags( in main()
117 toco::port::InitGoogle(argv[0], effective_argc, &effective_argv, true); in main()
118 toco::ToolMain(parsed_toco_flags, parsed_model_flags); in main()
|
| D | args.h | 32 namespace toco {
85 class Arg<toco::IntList> final {
112 const toco::IntList& value() const { return parsed_value_; } in value()
115 toco::IntList parsed_value_;
120 class Arg<toco::StringMapList> final {
179 const toco::StringMapList& value() const { return parsed_value_; } in value()
182 toco::StringMapList parsed_value_;
200 Arg<toco::IntList> input_shape;
201 Arg<toco::StringMapList> rnn_states;
202 Arg<toco::StringMapList> model_checks;
|
| D | toco_tooling.cc | 34 namespace toco { namespace
123 bool IsRealValued(toco::ArrayDataType type) { in IsRealValued()
124 return static_cast<bool>(type == toco::ArrayDataType::kFloat || in IsRealValued()
125 type == toco::ArrayDataType::kUint8); in IsRealValued()
178 model = toco::tflite::Import(model_flags, input_file_contents); in Import()
213 {new toco::RemoveUnusedOp}); in Transform()
239 transformations.Add(new toco::SplitLstmCellInputs); in Transform()
241 transformations.Add(new toco::MergeLstmCellInputs); in Transform()
314 toco::tflite::Export(model, allow_custom_ops, output_file_contents); in Export()
|
| D | toco_cmdline_flags.cc | 28 namespace toco { namespace
128 namespace port = toco::port; in ReadTocoFlagsFromCommandLineFlags()
185 toco::IODataType input_type; in ReadTocoFlagsFromCommandLineFlags()
186 QCHECK(toco::IODataType_Parse(parsed_toco_flags.input_type.value(), in ReadTocoFlagsFromCommandLineFlags()
204 toco::IODataType input_type; in ReadTocoFlagsFromCommandLineFlags()
205 QCHECK(toco::IODataType_Parse(input_types[0], &input_type)); in ReadTocoFlagsFromCommandLineFlags()
|
| D | BUILD | 263 # exposed by the :toco command-line tool.
298 "//tensorflow/contrib/lite/toco/tensorflow_graph_matching:resolve_cluster",
299 "//tensorflow/contrib/lite/toco/tflite:export",
300 "//tensorflow/contrib/lite/toco/tflite:import",
346 # :toco is the main public command-line tool exposing the functionality
349 name = "toco",
350 srcs = ["toco.cc"],
|
| D | toco_port.cc | 21 namespace toco { namespace
44 namespace toco { namespace
63 toco::port::file::Options ToOptions(const ::file::Options& options) { in ToOptions()
116 namespace toco { namespace
|
| /external/tensorflow/tensorflow/contrib/lite/toco/python/ |
| D | BUILD | 13 "//tensorflow/contrib/lite/toco:model_flags_proto_cc",
14 "//tensorflow/contrib/lite/toco:toco_flags_proto_cc",
15 "//tensorflow/contrib/lite/toco:toco_port",
16 "//tensorflow/contrib/lite/toco:toco_tooling",
24 srcs = ["toco.i"],
27 "//tensorflow/contrib/lite/toco:model_flags_proto_cc",
28 "//tensorflow/contrib/lite/toco:toco_flags_proto_cc",
58 "//tensorflow/contrib/lite/toco:model_flags_proto_py",
59 "//tensorflow/contrib/lite/toco:toco_flags_proto_py",
|
| D | toco_python_api.cc | 26 namespace toco { namespace
66 toco::ModelFlags model_flags; in TocoConvert()
70 toco::TocoFlags toco_flags; in TocoConvert()
74 std::unique_ptr<toco::Model> model = in TocoConvert()
75 toco::Import(toco_flags, model_flags, input_contents_txt); in TocoConvert()
76 toco::Transform(toco_flags, model.get()); in TocoConvert()
|
| D | toco_from_protos_test.py | 23 from tensorflow.contrib.lite.toco import model_flags_pb2
24 from tensorflow.contrib.lite.toco import toco_flags_pb2
25 from tensorflow.contrib.lite.toco import types_pb2
|
| /external/tensorflow/tensorflow/contrib/lite/python/ |
| D | lite.py | 36 from tensorflow.contrib.lite.toco import model_flags_pb2 as _model_flags_pb2
37 from tensorflow.contrib.lite.toco import toco_flags_pb2 as _toco_flags_pb2
38 from tensorflow.contrib.lite.toco import types_pb2 as _types_pb2
178 toco = _toco_flags_pb2.TocoFlags()
179 toco.input_format = input_format
180 toco.output_format = output_format
181 toco.drop_control_dependency = drop_control_dependency
183 toco.inference_type = inference_type
205 toco.inference_input_type = tflite_input_type
211 toco.SerializeToString(),
|
| D | BUILD | 11 # "//tensorflow/contrib/lite/toco/python:toco_from_protos",
17 "//tensorflow/contrib/lite/toco:model_flags_proto_py",
18 "//tensorflow/contrib/lite/toco:toco_flags_proto_py",
19 "//tensorflow/contrib/lite/toco/python:tensorflow_wrap_toco",
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/tests/ |
| D | BUILD | 14 "//tensorflow/contrib/lite/toco:graph_transformations",
15 "//tensorflow/contrib/lite/toco:model",
16 "//tensorflow/contrib/lite/toco:tooling_util",
25 "//tensorflow/contrib/lite/toco:graph_transformations",
26 "//tensorflow/contrib/lite/toco:model",
27 "//tensorflow/contrib/lite/toco:tooling_util",
|
| D | resolve_constant_concatenation_test.cc | 26 namespace toco { namespace
136 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>(); in PrepareModel()
139 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>().data.data(); in PrepareModel()
169 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
175 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
188 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
194 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
207 graph_transformation_set.Add(new toco::ResolveConstantConcatenation); in TEST_F()
213 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/ |
| D | BUILD | 19 "//tensorflow/contrib/lite/toco:toco_port",
33 "//tensorflow/contrib/lite/toco:model",
34 "//tensorflow/contrib/lite/toco:tooling_util",
51 "//tensorflow/contrib/lite/toco:model",
52 "//tensorflow/contrib/lite/toco:toco_port",
53 "//tensorflow/contrib/lite/toco:tooling_util",
87 "//tensorflow/contrib/lite/toco:tooling_util",
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| D | BUILD | 27 "//tensorflow/contrib/lite/toco:model",
41 "//tensorflow/contrib/lite/toco:tooling_util",
58 "//tensorflow/contrib/lite/toco:model",
87 "//tensorflow/contrib/lite/toco:model",
88 "//tensorflow/contrib/lite/toco:tooling_util",
119 "//tensorflow/contrib/lite/toco:model",
120 "//tensorflow/contrib/lite/toco:tooling_util",
|
| D | types.cc | 17 namespace toco { namespace
25 using NativeT = ::toco::DataType<T>; in CopyBuffer()
34 using NativeT = ::toco::DataType<T>; in CopyBuffer()
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | build_def.bzl | 127 toco = "//tensorflow/contrib/lite/toco:toco"
138 % (toco, src, out, options),
139 tools= [toco],
195 toco = "//tensorflow/contrib/lite/toco:toco"
202 cmd = ("$(locations :generate_examples) --toco $(locations %s) " % toco
208 toco,
|
| /external/tensorflow/tensorflow/contrib/lite/toco/g3doc/ |
| D | cmdline_examples.md | 24 //tensorflow/contrib/lite/toco:toco -- \
76 //tensorflow/contrib/lite/toco:toco -- \
103 //tensorflow/contrib/lite/toco:toco -- \
131 //tensorflow/contrib/lite/toco:toco -- \
179 //tensorflow/contrib/lite/toco:toco -- \
208 //tensorflow/contrib/lite/toco:toco -- \
236 //tensorflow/contrib/lite/toco:toco -- \
287 //tensorflow/contrib/lite/toco:toco -- \
309 //tensorflow/contrib/lite/toco:toco -- \
377 //tensorflow/contrib/lite/toco:toco -- \
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| D | graph_transformations.cc | 28 namespace toco { namespace
191 PrintModelStats(toco::port::StringF("Before %s", msg), *model); in RunGraphTransformations()
197 toco::port::StringF("After %s pass %d", msg, pass_index); in RunGraphTransformations()
|
| D | resolve_strided_slice_attributes.cc | 20 namespace toco { namespace
36 if (toco::RequiredBufferSizeForShape(start_array.shape()) > 4) { in Run()
|
| D | convert_trivial_transpose_to_reshape.cc | 22 namespace toco { namespace
61 string shape_array_name = toco::AvailableArrayName(*model, perm_array_name); in Run()
|
| D | convert_expanddims_to_reshape.cc | 26 namespace toco { namespace
76 string shape_array_name = toco::AvailableArrayName(*model, axis_array_name); in Run()
|
| /external/tensorflow/tensorflow/tools/ci_build/linux/cpu/ |
| D | run_py3_contrib.sh | 43 //tensorflow/contrib/lite/toco:toco \
|
| /external/tensorflow/tensorflow/tools/pip_package/ |
| D | BUILD | 159 "//tensorflow/contrib/lite/toco:toco",
160 "//tensorflow/contrib/lite/toco/python:toco_wrapper",
161 "//tensorflow/contrib/lite/toco/python:toco_from_protos",
|