Search Options

Results per page
Sort
Preferred Languages
Advance

Results 171 - 180 of 271 for getOperation (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

        registry.insert<mlir::tf_saved_model::TensorFlowSavedModelDialect,
                        ml_program::MLProgramDialect>();
      }
    
      void runOnOperation() override {
        if (failed(convertTFGlobals(getOperation()))) {
          signalPassFailure();
        }
      }
    };
    
    std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass() {
      return std::make_unique<LowerGlobalsToMlProgram>();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc

        }
      }
      return success();
    }
    
    void LiftQuantizableSpotsAsFunctionsPass::runOnOperation() {
      MLIRContext* ctx = &getContext();
      RewritePatternSet patterns(ctx);
      ModuleOp module_op = getOperation();
    
      simple_patterns::populateWithGenerated(patterns);
      fusion_patterns::populateWithGenerated(patterns);
      FrozenRewritePatternSet frozen_patterns(std::move(patterns));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc

          if (dequantize_op.use_empty()) {
            dequantize_op.erase();
          }
        }
      }
      return success();
    }
    
    void ModifyIONodesPass::runOnOperation() {
      auto func = getOperation();
      auto attrs = func->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function");
    
      // Handle the entry functions only.
      if (func.getName() != "main" && (!attrs || attrs.empty())) {
        return;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/pin_ops_with_side_effects.cc

     private:
      // This will be used recursively.
      const std::function<bool(Operation *)> op_has_side_effects_;
    };
    
    void PinOpsWithSideEffectsPass::runOnOperation() {
      auto fn = getOperation();
      // We're assuming (and checking) that there are no tfl::ControlNodeOps present
      // before this pass. We could relax this requirement by defining folding logic
      // for them.
      if (fn.walk([&](TFL::ControlNodeOp) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

    class XlaCallModuleSerializationPass
        : public impl::XlaCallModuleSerializationPassBase<
              XlaCallModuleSerializationPass> {
     public:
      void runOnOperation() override {
        mlir::ModuleOp module = getOperation();
        mlir::SymbolTableCollection symbol_table;
    
        mlir::WalkResult result =
            module.walk([&](mlir::TF::XlaCallModuleOp xla_call_module) {
              if (failed(SerializeXlaCallModule(symbol_table, module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  6. plugin/pkg/admission/certificates/ctbattest/admission_test.go

    	return t.obj
    }
    
    func (t *testAttributes) GetOldObject() runtime.Object {
    	return t.oldObj
    }
    
    func (t *testAttributes) GetName() string {
    	return t.name
    }
    
    func (t *testAttributes) GetOperation() admission.Operation {
    	return t.operation
    }
    
    func (t *testAttributes) GetUserInfo() user.Info {
    	return &user.DefaultInfo{Name: "ignored"}
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Wed Apr 24 18:25:29 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc

      builder.create<func::ReturnOp>(
          NameLoc::get(builder.getStringAttr(kTfQuantSaveReturnOpName)));
    }
    
    void InsertSaveOpPass::runOnOperation() {
      ModuleOp module_op = getOperation();
    
      func::FuncOp session_init_func = GetInitializerFunction(
          module_op, /*initializer_type=*/kTfSavedModelInitializerRestoreType);
      if (!session_init_func) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_weight.cc

      }
    };
    
    // TODO: b/264218457 - Refactors the current file to parse preset quantization
    // options and allow modular control of quantization specs.
    void QuantizeWeightPass::runOnOperation() {
      func::FuncOp func = getOperation();
      MLIRContext* ctx = func.getContext();
      RewritePatternSet patterns(ctx);
    
      patterns.add<QuantizeWeight>(ctx, quantization_component_spec_);
    
      FrozenRewritePatternSet frozen_patterns(std::move(patterns));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

                  << key;
              break;
            }
          }
        }
        return attrs;
      }
    };
    
    void TflToStablehloPass::runOnOperation() {
      func::FuncOp fn = getOperation();
      OpBuilder builder(fn.getContext());
      fn.walk([&](TFL::CustomOp custom_op) {
        builder.setInsertionPoint(custom_op);
        const uint8_t* option_buf = reinterpret_cast<const uint8_t*>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

                  << " outside compilation regions.";
      }
    }
    
    void MarkOpsForOutsideCompilation::runOnOperation() {
      auto module = getOperation();
      const Dialect* tf_dialect = getContext().getLoadedDialect("tf");
      if (!tf_dialect) {
        getOperation().emitError() << "'tf' dialect is not registered";
        return signalPassFailure();
      }
      RewritePatternSet patterns(&getContext());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
Back to top