- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 208 for tf_saved_model (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/dag_object_graph.py
class TestModule(tf.Module): def __init__(self): super(TestModule, self).__init__() self.child1 = Child() self.child2 = self.child1 # CHECK: tf_saved_model.global_tensor # CHECK-SAME: tf_saved_model.exported_names = ["child1.my_variable", "child2.my_variable"] if __name__ == '__main__':
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/keras.py
def __init__(self): super(TestModule, self).__init__() self.model = mnist_model() # CHECK: func {{@[a-zA-Z_0-9]+}}(%arg0: tensor<1x28x28x1xf32> {tf._user_specified_name = "x", tf_saved_model.index_path = [0]} # CHECK: attributes {{.*}} tf_saved_model.exported_names = ["my_predict"] @tf.function(input_signature=[ tf.TensorSpec([1, 28, 28, 1], tf.float32), ]) def my_predict(self, x): return self.model(x)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_initialized_variables_test_pass.cc
} void runOnOperation() override { // Pass an invalid session argument, which is a nullptr. if (failed(mlir::tf_saved_model::MarkInitializedVariablesInFunction( getOperation(), /*session=*/nullptr))) return signalPassFailure(); } }; } // namespace namespace tf_saved_model { static PassRegistration<MarkInitializedVariablesTestPass> mark_initialized_variables_test_pass;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
if (auto session_initializer = tf_saved_model::GetSessionInitializerOp(module)) { // clang-format off // Skip session initializer functions which are present in saved models. // For example: // "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> () // NOLINT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver_with_skipping.mlir
func.func @main(%arg0: tensor<10x1x1024xf32> {tf_saved_model.index_path = ["input_tensor"]}) -> (tensor<10x1x3xf32> {tf_saved_model.index_path = ["output"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input_tensor:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc
#include "tensorflow/compiler/mlir/lite/transforms/passes.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" namespace mlir { namespace TFL { namespace { #define GEN_PASS_DEF_INSERTCALLONCEOPFROMSESSIONINITIALIZERPASS #include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc" // This pass inserts a TFL::CallOnce op when tf_saved_model's session // initializer is given. class InsertCallOnceOpFromSessionInitializerPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
#include "mlir/IR/UseDefLists.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.h" #include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device_mgr.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/compiler/mlir/utils/name_utils.h" namespace mlir { namespace quant { namespace { using ::mlir::tf_saved_model::GetInitializerFunction; using ::mlir::tf_saved_model::GetSessionInitializerOp; using ::mlir::tf_saved_model::kTfSavedModelExportedNamesAttr; using ::mlir::tf_saved_model::kTfSavedModelInitializerRestoreType;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
#include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/core/platform/path.h" namespace mlir { namespace tf_saved_model { namespace { #define GEN_PASS_DEF_FREEZEASSETSPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
#include "mlir/Pass/PassRegistry.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h" namespace mlir { namespace tf_saved_model { namespace { #define GEN_PASS_DEF_ADDFUNCTIONSFOREXPORTEDNAMESPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0)