- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for init_ops (0.36 sec)
-
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.h
#include "pybind11/stl.h" // from @pybind11 namespace py = pybind11; void init_basic_classes(py::module& m); void init_types(py::module& m); void init_builders(py::module& m); void init_ops(py::module& m); void init_attrs(py::module& m);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 08:41:42 UTC 2023 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.cc
module->emitError("Invalid MLIR module: failed verification."); return false; } return true; }); init_basic_classes(m); init_types(m); init_builders(m); init_ops(m); init_attrs(m);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 08:41:42 UTC 2023 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
for (auto init_op : init_table_from_text_file_ops_to_erase) { // Replace the InitializeTableFromTextFileV2Op to use the saved model's // asset filepath. builder.setInsertionPoint(init_op); builder.create<TF::InitializeTableFromTextFileV2Op>( init_op.getLoc(), init_op.getTableHandle(), const_op.getResult(), init_op.getKeyIndex(), init_op.getValueIndex(),
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/tests/tf_saved_model/common_v1.py
signature_def_map, init_op, assets_collection = create_signature() sess = tf.Session() sess.run(tf.initializers.global_variables()) builder = tf.saved_model.builder.SavedModelBuilder(save_model_path) builder.add_meta_graph_and_variables( sess, [tf.saved_model.tag_constants.SERVING], signature_def_map, main_op=init_op, assets_collection=assets_collection,
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/aot/tests/make_test_graphs.py
y = variable_v1.VariableV1(constant_op.constant([0]), name='y_saved') math_ops.add(x, y, name='x_y_sum') init_op = variables.global_variables_initializer() saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1) with session.Session() as sess: sess.run(init_op) sess.run(y.assign(y + 42)) # Without the checkpoint, the variable won't be set to 42.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 15 15:25:23 UTC 2023 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
ModuleOp module = getOperation(); if (!HasTfSavedModelSemantics(module)) { return; } auto init_op = GetSessionInitializerOp(module); if (init_op == nullptr || init_op.getInitializers().empty()) { return; } SymbolTable symbol_table(module); for (auto initializer : init_op.getInitializers()) { auto func = symbol_table.lookup<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/examples/mnist/BUILD
], licenses = ["notice"], ) package_group( name = "friends", packages = [ "//tensorflow/compiler/mlir/tfr/...", ], ) gen_op_libraries( name = "mnist_ops", src = "ops_defs.py", deps = [ "//tensorflow:tensorflow_py", "//tensorflow/python/framework:ops", "//tensorflow/python/ops:math_ops", "//tensorflow/python/ops:math_ops_gen",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 24 11:50:40 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_test.cc
"tf_saved_model.session_initializer"() {initializers = [@init_func]} : () -> () func.func @init_func() attributes {tf_saved_model.exported_names = ["init_func"], tf_saved_model.initializer_type = "init_op"} { func.return } } )mlir"; Block block; ModuleOp module_op = ParseModuleOp(kModuleOpStr, block, ctx_); func::FuncOp init_func_op = GetInitializerFunction(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_remove_vars_in_session_initializer.mlir
// Test case: Variable removed for the initializer function whose // initializer_type is "init_op". "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> () func.func @init() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer"], tf_saved_model.initializer_type = "init_op"} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 00:22:36 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/convert_session_initializer_to_function.mlir
"tf_saved_model.session_initializer"() {initializers = [@init1]} : () -> () func.func @init1() attributes {tf_saved_model.exported_names = ["init1"], tf_saved_model.initializer_type = "init_op"} { %0 = "tf.Const"() {value = dense<42> : tensor<1xi64>} : () -> tensor<1xi64> return } } // ----- // CHECK-LABEL: multiple_initializers // CHECK-NOT: tf_saved_model.session_initializer
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 02 16:24:00 UTC 2022 - 1.9K bytes - Viewed (0)