- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 208 for tf_saved_model (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.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" namespace mlir { namespace tf_saved_model { namespace { using mlir::TF::VarHandleOp; #define GEN_PASS_DEF_REMOVEVARIABLESINSESSIONINITIALIZERPASS
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" #include "tensorflow/compiler/mlir/tensorflow/translate/import_model.h" namespace mlir { namespace quant { namespace { using ::mlir::tf_executor::FetchOp; using ::mlir::tf_executor::GraphOp; using ::mlir::tf_executor::IslandOp; using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr; using ::tensorflow::kImportModelDefaultGraphFuncName;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
#include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" namespace mlir { std::vector<llvm::StringRef> GetEntryFunctionAttributeNames() { return {"tf.entry_function", tf_saved_model::kTfSavedModelInitializerTypeAttr}; } bool IsEntryFunction(func::FuncOp func) { for (const auto &attr : GetEntryFunctionAttributeNames()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/defun_export.py
# has a corresponding tf.LegacyCall op. # CHECK: func {{@[a-zA-Z_0-9]+}}( # CHECK-SAME: [[ARG0:%.*]]: tensor<3x1xf32> {tf_saved_model.index_path = ["y"]}, # CHECK-SAME: [[ARG1:%.*]]: tensor<3x1xf32> {tf_saved_model.index_path = ["x"]} # # CHECK-NEXT: [[R0:%.*]] = "tf.LegacyCall"([[ARG1]], [[ARG0]]) z = plus(x, y) tensor_info_x = tf.compat.v1.saved_model.utils.build_tensor_info(x)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver.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: Thu Apr 25 01:09:50 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.h
#include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "tensorflow/core/public/session.h" namespace mlir { namespace tf_saved_model { // Freezes readonly variables in the graph. LogicalResult FreezeVariables(ModuleOp module, tensorflow::Session* session); } // namespace tf_saved_model } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 14:25:57 UTC 2022 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/remove_var_init_by_const.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" namespace mlir { namespace quant { namespace { using ::mlir::tf_saved_model::GetInitializerFunction; using ::mlir::tf_saved_model::kTfSavedModelInitializerRestoreType; // A pass that removes `tf.AssignVariableOp(tf.VarHandleOp, tf.Const)` patterns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 03 12:04:03 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composite_functions_xla.mlir
func.func @conv_with_two_layers(%arg0: tensor<1x3x4x3xf32> {tf_saved_model.index_path = ["input"]}) -> (tensor<1x3x2x2xf32> {tf_saved_model.index_path = ["output"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 08 01:16:10 UTC 2024 - 25.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.h
#include "mlir/Pass/Pass.h" // from @llvm-project namespace mlir { namespace tf_saved_model { // Create a pass that removes function arguments that map to global tensors. std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass(); // Register this pass in the global registry of MLIR. void RegisterLowerGlobalsToMlProgramPass(); } // namespace tf_saved_model } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 14:17:18 UTC 2022 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir
// RUN: tf-tfrt-opt -verify-diagnostics -split-input-file -tfrt-fuse-tpu-compile-and-execute-ops -canonicalize %s | FileCheck %s --dump-input=fail --dump-input-filter=all module attributes {tf_saved_model.semantics} { // Test fusing _TPUCompileMlirOp and TPUExecuteOp into TPUCompileMlirAndExecuteOp. // CHECK-LABEL: func private @test_fuse_tpu_ops func.func private @test_fuse_tpu_ops(%arg0: tensor<*xi32>, %arg1: tensor<*x!tf_type.resource>) -> tensor<*xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.8K bytes - Viewed (0)