Home
last modified time | relevance | path

Searched refs:supported_ops (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc74 llvm::DenseSet<OperationName>* supported_ops) { in AddSupportedOpsUsingFolding() argument
84 supported_ops->insert(allowlist_ops.begin(), allowlist_ops.end()); in AddSupportedOpsUsingFolding()
90 llvm::DenseSet<OperationName>* supported_ops) { in AddSupportedControlFlowOps() argument
91 supported_ops->insert( in AddSupportedControlFlowOps()
93 supported_ops->insert( in AddSupportedControlFlowOps()
95 supported_ops->insert( in AddSupportedControlFlowOps()
101 llvm::DenseSet<OperationName>* supported_ops) { in AddRewrittenEmbeddingOps() argument
102 supported_ops->insert(OperationName( in AddRewrittenEmbeddingOps()
104 supported_ops->insert(OperationName( in AddRewrittenEmbeddingOps()
113 llvm::DenseSet<OperationName>* supported_ops) { in AddRewrittenCompositeOps() argument
[all …]
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer_test.py125 supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS]
126 converter.target_spec.supported_ops = supported_ops
168 supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS]
169 converter.target_spec.supported_ops = supported_ops
206 supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS]
207 converter.target_spec.supported_ops = supported_ops
/external/tensorflow/tensorflow/lite/tools/optimize/python/
Dmodify_model_interface_lib_test.py32 supported_ops=tf.lite.OpsSet.TFLITE_BUILTINS_INT8): argument
50 converter.target_spec.supported_ops = [supported_ops]
101 supported_ops=tf.lite.OpsSet
/external/tensorflow/tensorflow/lite/python/
Dlite_flex_test.py60 converter.target_spec.supported_ops = set([lite.OpsSet.SELECT_TF_OPS])
91 self.assertEqual(converter.target_spec.supported_ops,
128 converter.target_spec.supported_ops = set([lite.OpsSet.SELECT_TF_OPS])
189 converter.target_spec.supported_ops = set([lite.OpsSet.SELECT_TF_OPS])
208 converter.target_spec.supported_ops = set([lite.OpsSet.SELECT_TF_OPS])
Dlite.py196 supported_ops=None, argument
199 if supported_ops is None:
200 supported_ops = {OpsSet.TFLITE_BUILTINS}
201 self.supported_ops = supported_ops
384 self._target_spec.supported_ops)) or (set(
389 in set(self._target_spec.supported_ops))
393 self._target_spec.supported_ops)) or (OpsSet.SELECT_TF_OPS in set(
394 self._target_spec.supported_ops))
464 set([OpsSet.SELECT_TF_OPS]) == set(self.target_spec.supported_ops))
513 "target_ops": self.target_spec.supported_ops,
[all …]
Dlite_v2_test.py248 quantized_converter.target_spec.supported_ops = [
253 quantized_converter.target_spec.supported_ops = [
258 quantized_converter.target_spec.supported_ops = [
293 quantized_converter.target_spec.supported_ops = [
317 converter.target_spec.supported_ops = [lite.OpsSet.TFLITE_BUILTINS_INT8]
415 quantized_converter.target_spec.supported_ops = [
539 quantized_converter.target_spec.supported_ops = [
545 quantized_converter.target_spec.supported_ops = [
550 quantized_converter.target_spec.supported_ops = [
557 quantized_converter.target_spec.supported_ops = [
[all …]
Dtflite_convert.py190 converter.target_spec.supported_ops = set()
195 converter.target_spec.supported_ops.add(lite.OpsSet(option))
198 if lite.OpsSet.SELECT_TF_OPS not in converter.target_spec.supported_ops:
Dutil_test.py307 converter.target_spec.supported_ops = {tf.lite.OpsSet.TFLITE_BUILTINS_INT8}
309 converter.target_spec.supported_ops = {
Dlite_test.py347 quantized_converter.target_spec.supported_ops = [
353 quantized_converter.target_spec.supported_ops = [
358 quantized_converter.target_spec.supported_ops = [
364 quantized_converter.target_spec.supported_ops = [
928 def testQuantizeInt8And16x8(self, supported_ops, enable_mlir_converter): argument
945 quantized_converter.target_spec.supported_ops = supported_ops
1215 quantized_converter.target_spec.supported_ops = [
1610 converter.target_spec.supported_ops = [
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc227 std::vector<std::string> supported_ops = toco::GetPotentiallySupportedOps(); in TocoGetPotentiallySupportedOps() local
228 PyObject* list = PyList_New(supported_ops.size()); in TocoGetPotentiallySupportedOps()
229 for (size_t i = 0; i < supported_ops.size(); ++i) { in TocoGetPotentiallySupportedOps()
230 const std::string& op = supported_ops[i]; in TocoGetPotentiallySupportedOps()
/external/tensorflow/tensorflow/lite/testing/
Dtoco_convert.py140 converter.target_spec.supported_ops = [
145 converter.target_spec.supported_ops = [
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lite.-target-spec.pbtxt7 …argspec: "args=[\'self\', \'supported_ops\', \'supported_types\', \'experimental_select_user_tf_op…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lite.-target-spec.pbtxt7 …argspec: "args=[\'self\', \'supported_ops\', \'supported_types\', \'experimental_select_user_tf_op…
/external/kernel-headers/original/uapi/linux/
Dkeyctl.h99 __u32 supported_ops; /* Which ops are supported */ member
/external/tensorflow/tensorflow/lite/micro/testing/
Dgenerate_test_models.py65 converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS_INT8]
/external/tensorflow/tensorflow/lite/testing/model_coverage/
Dmodel_coverage_lib.py140 converter.target_spec.supported_ops = kwargs["target_ops"]
148 converter.target_spec.supported_ops = [_lite.OpsSet.TFLITE_BUILTINS_INT8]
155 converter.target_spec.supported_ops = \
Dmodel_coverage_quantization_test.py111 converter.target_spec.supported_ops = [
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_dispatch_test.py711 supported_ops = [
754 for element in supported_ops + supported_ops_v1:
760 for element in supported_ops + supported_ops_v2:
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dquantize.cc47 static const std::set<OperatorType> supported_ops{ in SupportsQuantization() local
111 return supported_ops.find(type) != supported_ops.end(); in SupportsQuantization()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_device_selection_test.cc936 void SetNodeSupport(bool* supported_ops) { in SetNodeSupport() argument
938 supported_ops[i] = (unsupported_indexes_.count(i) == 0); in SetNodeSupport()
990 bool* supported_ops) -> int { in Init() argument
992 supported_ops); in Init()
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dapi_updates.md28 * Renamed the `target_ops` attribute to `target_spec.supported_ops`
/external/tensorflow/tensorflow/lite/g3doc/performance/
Dpost_training_quantization.md121 <b>converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS_INT8]</b>
175 converter.target_spec.supported_ops = [tf.lite.OpsSet.EXPERIMENTAL_TFLITE_BUILTINS_ACTIVATIONS_INT1…
187 converter.target_spec.supported_ops = [tf.lite.OpsSet.EXPERIMENTAL_TFLITE_BUILTINS_ACTIVATIONS_INT1…
/external/tensorflow/tensorflow/lite/experimental/quantization_debugger/
Ddebugger_test.py64 converter.target_spec.supported_ops = [lite.OpsSet.TFLITE_BUILTINS_INT8]
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dngrams_test.py103 converter.target_spec.supported_ops = [
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dop_select_allowlist.md722 converter.target_spec.supported_ops = [
747 converter.target_spec.supported_ops = [

12