Searched refs:unfold_batch_matmul (Results 1 – 6 of 6) sorted by relevance
36 unfold_batch_matmul(true),58 bool unfold_batch_matmul; member
95 pass_config.unfold_batch_matmul = false; in ConvertGraphDefToTFLiteFlatBuffer()
182 pass_config.unfold_batch_matmul = false; in ConvertSavedModelToTFLiteFlatBuffer()
44 bool unfold_batch_matmul, bool allow_bf16_and_f16_type_legalization);
86 explicit PrepareTFPass(bool unfold_batch_matmul, in PrepareTFPass() argument88 unfold_batch_matmul_ = unfold_batch_matmul; in PrepareTFPass()1495 bool unfold_batch_matmul, bool allow_bf16_type_legalization) { in CreatePrepareTFPass() argument1496 return std::make_unique<PrepareTFPass>(unfold_batch_matmul, in CreatePrepareTFPass()
195 pass_config.unfold_batch_matmul, in AddTFToTFLConversionPasses()