- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for DEFINE_string (0.27 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/gen_quantized_function_library.py
from absl import app from absl import flags # TODO(b/263048929): Create a test for gen_quantized_function_library. _OUTPUT_FILE = flags.DEFINE_string('output_file', None, 'output file location') _SRCS = flags.DEFINE_string('src', None, 'source file locations') _NAMESPACE = flags.DEFINE_string('namespace', 'mlir::quant', 'namespace in the generated file') flags.mark_flags_as_required(['output_file', 'src'])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 20 01:38:06 UTC 2022 - 8.4K bytes - Viewed (0) -
tensorflow/cc/saved_model/testdata/generate_chunked_models.py
from tensorflow.python.util import compat from tensorflow.tools.proto_splitter import constants from tensorflow.tools.proto_splitter.python import saved_model as proto_splitter SPLITTER_TESTDATA_PATH = flags.DEFINE_string( "path", None, help="Path to testdata directory.") def generate_non_chunked_model(non_chunked_dir: str): root = module.Module() root.c = constant_op.constant(np.random.random_sample([150, 150]))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 21:43:11 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/common.py
from tensorflow.python import pywrap_mlir # pylint: disable=g-direct-tensorflow-import from tensorflow.python.lib.io import file_io # Use /tmp to make debugging the tests easier (see README.md) flags.DEFINE_string('save_model_path', '', 'Path to save the model to.') FLAGS = flags.FLAGS # This function needs to take a "create_module_fn", as opposed to just the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 23:49:27 UTC 2023 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/common_v1.py
from absl import logging import tensorflow.compat.v1 as tf from tensorflow.python import pywrap_mlir # pylint: disable=g-direct-tensorflow-import # Use /tmp to make debugging the tests easier (see README.md) flags.DEFINE_string('save_model_path', '', 'Path to save the model to.') FLAGS = flags.FLAGS def set_tf_options(): # Default TF1.x uses reference variables that are not supported by SavedModel
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 20 13:19:26 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
from tensorflow.python.ops import gen_nn_ops from tensorflow.python.ops import math_ops from tensorflow.python.platform import flags Composite = composite.Composite FLAGS = flags.FLAGS flags.DEFINE_string( 'output', None, 'Path to write the genereated register op file and MLIR file.') flags.DEFINE_bool('gen_register_op', True, 'Generate register op cc file or tfr mlir file.')
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0)