Home
last modified time | relevance | path

Searched refs:unfold_batchmatmul (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dtoco_cmdline_flags.cc195 Flag("unfold_batchmatmul", parsed_flags.unfold_batchmatmul.bind(), in ParseTocoFlagsFromCommandLineFlags()
196 parsed_flags.unfold_batchmatmul.default_value(), ""), in ParseTocoFlagsFromCommandLineFlags()
301 READ_TOCO_FLAG(unfold_batchmatmul, FlagRequirement::kNone); in ReadTocoFlagsFromCommandLineFlags()
Dargs.h196 Arg<bool> unfold_batchmatmul = Arg<bool>(true); member
Dtoco_flags.proto242 optional bool unfold_batchmatmul = 35 [default = true]; field
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_translate_cl.h44 extern llvm::cl::opt<bool> unfold_batchmatmul;
Dtf_tfl_translate_cl.cc124 opt<bool> unfold_batchmatmul( variable
Dtf_tfl_translate.cc270 pass_config.unfold_batch_matmul = unfold_batchmatmul; in main()
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dunroll_batch_matmul_disabled.pbtxt1 # RUN: tf_tfl_translate -unfold_batchmatmul=false -tf-input-arrays=Placeholder,Placeholder_1 -tf-in…
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dgraphdef_to_tfl_flatbuffer.cc102 pass_config.unfold_batch_matmul = toco_flags.unfold_batchmatmul(); in ConvertGraphDefToTFLiteFlatBuffer()
Dsaved_model_to_tfl_flatbuffer.cc187 pass_config.unfold_batch_matmul = toco_flags.unfold_batchmatmul(); in ConvertSavedModelToTFLiteFlatBuffer()
Djax_to_tfl_flatbuffer.cc153 pass_config.unfold_batch_matmul = toco_flags.unfold_batchmatmul(); in ConvertJaxToTFLiteFlatBuffer()
/external/tensorflow/tensorflow/lite/python/
Dconvert.py488 unfold_batchmatmul=True, argument
627 conversion_flags.unfold_batchmatmul = unfold_batchmatmul