Search Options

Results per page
Sort
Preferred Languages
Advance

Results 181 - 190 of 271 for getOperation (0.15 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

            return WalkResult::interrupt();
          // Mark op as stateful due to side-effecting communication ops.
          if_op->setAttr("is_stateless", builder.getBoolAttr(false));
          MarkOutsideCompiled(host_if.getOperation());
        }
        if (auto while_op = llvm::dyn_cast<mlir::TF::WhileRegionOp>(op)) {
          if (!HasOutsideCompilationNested(op)) return WalkResult::advance();
          OpBuilder builder(while_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h

                                  int activation_number_of_bits = 8) {
      if (llvm::isa<TFL::LSTMOp>(op.getOperation())) {
        lstm_variant->op_code = tflite::BuiltinOperator_LSTM;
      } else if (llvm::isa<TFL::UnidirectionalSequenceLSTMOp>(op.getOperation())) {
        lstm_variant->op_code =
            tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM;
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 28K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/debug/debug_test.cc

     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MutatePass)
    
      void runOnOperation() override {
        mlir::OpBuilder builder(&getContext());
        getOperation()->setAttr("tfl.random_attr", builder.getUnitAttr());
      }
    };
    
    class AlwaysFailPass
        : public mlir::PassWrapper<AlwaysFailPass, mlir::OperationPass<>> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 17 11:15:16 UTC 2024
    - 9.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc

          rewriter, input_val, result_type, unique_func_name, func_input_arg);
    
      // Fills the body.
      Operation* last_op_in_func =
          quantization_operations_func(rewriter, func_op.getOperation(),
                                       func_input_arg, result_type, quant_type);
    
      // Connect the function in the existing graph.
      auto end_call_op = FinalizeFunctionRegister(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

      }
    
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftTfliteFlexOpsPass)
    
      void runOnOperation() override {
        MLIRContext* context = &getContext();
        func::FuncOp func = getOperation();
    
        mlir::RewritePatternSet patterns(context);
        AddLiftTfliteFlexOpsPatterns(context, patterns);
        if (failed(applyPatternsAndFoldGreedily(func, std::move(patterns)))) {
          signalPassFailure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  6. staging/src/k8s.io/apiserver/pkg/admission/plugin/webhook/generic/webhook.go

    		} else if !matchResult.Matches {
    			admissionmetrics.Metrics.ObserveMatchConditionExclusion(ctx, h.GetName(), "webhook", h.GetType(), string(attr.GetOperation()))
    			// if no match, always skip webhook
    			return nil, nil
    		}
    	}
    
    	return invocation, nil
    }
    
    type attrWithResourceOverride struct {
    	admission.Attributes
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 10 22:07:40 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc

        XlaCallModuleDeserializationPassBase::getDependentDialects(registry);
        mlir::func::registerAllExtensions(registry);
      }
    
      void runOnOperation() override {
        ModuleOp module = getOperation();
        SymbolTableCollection symbol_tables;
        WalkResult result = module.walk([&](XlaCallModuleOp op) {
          if (failed(DeserializeXlaCallModule(&getContext(), symbol_tables, module,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc

    };
    
    static PassRegistration<QuantizeWeightsPass> pass;
    
    void QuantizeWeightsPass::runOnOperation() {
      MLIRContext* ctx = &getContext();
      auto module_op = getOperation();
      RewritePatternSet patterns(ctx);
    
      patterns.add<QuantizeConstWeights>(ctx, quant_options_);
    
      FrozenRewritePatternSet frozen_patterns(std::move(patterns));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 11.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc

      generator_op->replaceAllUsesWith(generator_region->getResults());
      generator_op->erase();
    
      return success();
    }
    
    void FunctionalControlFlowToRegions::runOnOperation() {
      ModuleOp module = getOperation();
      auto result = module.walk([&](Operation* op) {
        if (IfOp if_op = llvm::dyn_cast<IfOp>(op)) {
          if (failed(ConvertIfOp(if_op))) {
            op->emitOpError() << "failed to convert to region form";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc

          while_op.getResult(i).replaceAllUsesWith(new_while_op.getResult(j));
        }
      }
      while_op.erase();
      return erase_indices.any();
    }
    
    void ReduceWhileOperandsPass::runOnOperation() {
      auto fn = getOperation();
      fn.walk([&](TFL::WhileOp while_op) { ReduceWhileOperands(while_op); });
    }
    
    static PassRegistration<ReduceWhileOperandsPass> pass;
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
Back to top