- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for filename_tensor_name (0.2 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
const std::string filename_tensor_name = FindFilePrefixTensorName(graph_def); const std::string restore_op_name = GetNodeName(control_ret_node_names, kTfSavedModelInitializerRestoreType); const std::string save_node_name = GetNodeName(control_ret_node_names, kTfQuantSaveOpName); const std::vector<absl::string_view> fields = { filename_tensor_name, restore_op_name, save_node_name};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
ASSERT_TRUE(TextFormat::ParseFromString( R"pb(filename_tensor_name: "my_file")pb", &saver_def)); const ExportedModel exported_model = CreateExportedModelFromGraphDef( GraphDef(), /*init_node_name=*/"", /*checkpoint_dir=*/"", saver_def, /*function_aliases=*/{}, /*asset_file_defs=*/{}); EXPECT_THAT(exported_model.saver_def().filename_tensor_name(), "my_file"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h
// Creates a new `SaverDef` instance, which contains information regarding // checkpoint saving and restoring. This function returns a `SaverDef` instance // with four fields populated: `version`, `filename_tensor_name`, // `restore_op_name` and `save_tensor_name`. For valid quantized `graph_def` and // `control_ret_node_names`, it should be able to retrieve the last three fields // if there is at lest one variable in the graph. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/cc/saved_model/loader.cc
TF_RETURN_IF_ERROR(RunRestore(run_options, export_dir, meta_graph.saver_def().restore_op_name(), meta_graph.saver_def().filename_tensor_name(), asset_file_defs, session->get())); } // Record walltime spent in restoring graph from disk, but postpone metric // increments until graph init finishes.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 04:36:00 UTC 2024 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)