Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for removeResultAttr (0.13 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/drop_savedmodel_semantics.cc

              func.removeArgAttr(i, bound_input);
            }
            func.removeArgAttr(i, index_path);
          }
          for (unsigned i = 0, e = func.getNumResults(); i != e; ++i) {
            func.removeResultAttr(i, bound_input);
            func.removeResultAttr(i, index_path);
          }
        });
    
        // Clean up modules from tf_saved_model attributes.
        module->removeAttr("tf_saved_model.semantics");
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 13 23:39:10 UTC 2022
    - 3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/strip_saved_module_metadata.cc

            func.getResultAttrs(i),
            [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); }));
        for (auto namedAttr : stripAttrs) {
          func.removeResultAttr(i, namedAttr.getName());
        }
      }
    }
    
    void StripSavedModuleMetadataPass::runOnOperation() {
      auto module = getOperation();
      StripModule(module);
      module.walk(StripFunction);
    }
    
    }  // namespace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 08:06:04 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc

            func.getResultAttrs(i),
            [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); }));
        for (auto namedAttr : stripAttrs) {
          func.removeResultAttr(i, namedAttr.getName());
        }
      }
    }
    
    void StripTfAttributesPass::runOnOperation() {
      ModuleOp module = getOperation();
    
      // strip module itself
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 08:06:04 UTC 2023
    - 3.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc

          }
        }
      }
      for (int i = 0; i < f.getNumResults(); ++i) {
        for (NamedAttribute a : f.getResultAttrs(i)) {
          if (a.getName().strref().starts_with("tf_saved_model.")) {
            f.removeResultAttr(i, a.getName());
          }
        }
      }
    }
    
    void AddFunctionsForExportedNamesPass::runOnOperation() {
      Block& module_body = getOperation().getRegion().front();
      for (auto f :
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 19 08:06:04 UTC 2023
    - 4.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc

        for (auto& attr : iface.getResultAttrs(i)) {
          const StringAttr& attr_name = attr.getName();
          if (attr_name.getValue().starts_with("tf_saved_model.")) {
            func.removeResultAttr(i, attr_name);
          }
        }
      }
    }
    
    // Information to identify an output in its node and in the model output list.
    // Ex: If the model output list is ["add:0", "topk:0": "topk:1"], then the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.5K bytes
    - Viewed (0)
Back to top