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.cc507 const TensorFlowImportFlags& tf_import_flags) { in GetInputsCount() argument
508 if (tf_import_flags.drop_control_dependency) { in GetInputsCount()
519 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in CheckInputsCount() argument
521 if (GetInputsCount(node, tf_import_flags) != expected_input_count) { in CheckInputsCount()
566 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertConstOperator() argument
618 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertConvOperator() argument
621 TF_RETURN_IF_ERROR(CheckInputsCount(node, tf_import_flags, 2)); in ConvertConvOperator()
693 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertDepthwiseConvOperator() argument
696 TF_QCHECK_OK(CheckInputsCount(node, tf_import_flags, 2)); in ConvertDepthwiseConvOperator()
764 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags, in ConvertDepthToSpaceOperator() argument
[all …]
Dimport_tensorflow.h38 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
42 const ModelFlags& model_flags, const TensorFlowImportFlags& tf_import_flags,
Dtoco_tooling.cc211 TensorFlowImportFlags tf_import_flags; in Import() local
212 tf_import_flags.drop_control_dependency = in Import()
217 tf_import_flags.import_all_ops_as_unsupported = in Import()
220 model = ImportTensorFlowGraphDef(model_flags, tf_import_flags, in Import()
Dimport_tensorflow_test.cc46 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags,