Home
last modified time | relevance | path

Searched refs:tf_import_flags (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc541 const TensorFlowImportFlags& tf_import_flags) { in GetInputsCount() argument
542 if (tf_import_flags.drop_control_dependency) { in GetInputsCount()
553 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in CheckInputsCount() argument
555 if (GetInputsCount(node, tf_import_flags) != expected_input_count) { in CheckInputsCount()
686 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertUnsupportedOperator() argument
706 const int num_inputs = GetInputsCount(node, tf_import_flags); in ConvertUnsupportedOperator()
774 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertConstOperator() argument
830 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertConvOperator() argument
833 TF_RETURN_IF_ERROR(CheckInputsCount(node, tf_import_flags, 2)); in ConvertConvOperator()
911 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertDepthwiseConvOperator() argument
[all …]
Dimport_tensorflow.h39 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
45 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
Dtoco_tooling.cc216 TensorFlowImportFlags tf_import_flags; in Import() local
217 tf_import_flags.drop_control_dependency = in Import()
222 tf_import_flags.import_all_ops_as_unsupported = in Import()
225 model = ImportTensorFlowGraphDef(model_flags, tf_import_flags, in Import()
Dimport_tensorflow_test.cc47 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags,