Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for call_op (0.22 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc

        if (method.ok() && method->has_static_range_ptq()) return true;
      }
    
      TF::PartitionedCallOp call_op = dyn_cast_or_null<TF::PartitionedCallOp>(op);
      return call_op && call_op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) &&
             call_op->getAttrOfType<StringAttr>(kQuantTraitAttrName).getValue() ==
                 llvm::StringRef(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 14.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

      for (auto& function_use : function_uses.value()) {
        auto call_op = function_use.getUser();
        auto caller_func = call_op->getParentOfType<func::FuncOp>();
        if (!caller_func) return failure();
    
        builder.setInsertionPoint(call_op);
        for (auto [lifted_op, arg_idx] : lifted_op_and_arg_idx) {
          auto new_op = builder.clone(*lifted_op, mapping);
          call_op->insertOperands(arg_idx, new_op->getResult(0));
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc

      return method.has_weight_only_ptq();
    }
    
    bool IsWeightOnlyQuantizableOp(const Operation& op) {
      if (auto call_op = dyn_cast<TF::XlaCallModuleOp>(op)) {
        StringRef entry_function_name = GetEntryFunctionName(call_op);
        absl::StatusOr<Method> quantization_method = GetQuantizationMethod(call_op);
        return ContainsConvOrDot(entry_function_name) && quantization_method.ok() &&
               quantization_method->has_weight_only_ptq();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc

      func::FuncOp main_fn = FindMainFuncOp(*module_op);
      ASSERT_THAT(main_fn, NotNull());
    
      Operation* call_op = FindOperationOfType<TF::XlaCallModuleOp>(main_fn);
      EXPECT_FALSE(IsHybridQuantizedOp(call_op));
    }
    
    constexpr absl::string_view kModuleDotGeneralFullyConnected = R"mlir(
      module {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 22.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

            max_iterations);
      } else if (auto call_op = dyn_cast<CallOpInterface>(op)) {
        if (auto func =
                dyn_cast<func::FuncOp>(call_op.resolveCallable(&symbol_table_))) {
          PropagateConstantToCallee(call_op, func, module);
          FailureOr<bool> failure_or_converged = PropagateShapeToFunctions(
              module, call_op.getArgOperands().getTypes(), {func}, max_iterations);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      op_builder.setInsertionPointToStart(&region.front());
      auto call_op = op_builder.create<mlir::func::CallOp>(
          loc, func.getFunctionType().getResults(), func.getSymName(),
          region.getArguments());
      op_builder.create<mlir::TFL::YieldOp>(loc, call_op.getResults());
    }
    
    void InlineStablehloOpRegion(mlir::Region& region, mlir::func::FuncOp func) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      auto get_call_index = [&](mlir::Block& b) -> std::optional<int> {
        if (b.getOperations().size() != 2) return std::nullopt;
        if (auto call_op = dyn_cast<mlir::func::CallOp>(b.front()))
          return subgraph_index_map_.at(call_op.getCallee().str());
        return std::nullopt;
      };
      auto body_subgraph_index = get_call_index(op.getBody().front());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      auto inputs = func_ty.getInputs();
      block->addArguments(inputs, SmallVector<Location>(inputs.size(), loc));
      mlir::func::CallOp call_op = rewriter.create<mlir::func::CallOp>(
          loc, func, func_ty.getResults(), block->getArguments());
      rewriter.create<mhlo::ReturnOp>(loc, call_op.getResults());
    }
    
    //===----------------------------------------------------------------------===//
    // Op converters.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc

      // only executes ops with no side effects.
      mutable llvm::SmallDenseMap<Operation*, bool> is_pure_function_;
    };
    
    bool OpSideEffectCollector::IsCallToPureFunction(Operation* callOp) const {
      auto call = llvm::dyn_cast<CallOpInterface>(callOp);
      if (!call)
        return false;  // not a call
      func::FuncOp func_op = dyn_cast<func::FuncOp>(call.resolveCallable(
          &symbol_table_collection_));
      return IsPureFunction(func_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 41.2K bytes
    - Viewed (0)
  10. src/runtime/testdata/testprogcgo/coro.go

    // license that can be found in the LICENSE file.
    
    //go:build goexperiment.rangefunc && !windows
    
    package main
    
    /*
    #include <stdint.h> // for uintptr_t
    
    void go_callback_coro(uintptr_t handle);
    
    static void call_go(uintptr_t handle) {
    	go_callback_coro(handle);
    }
    */
    import "C"
    
    import (
    	"fmt"
    	"iter"
    	"runtime/cgo"
    )
    
    func init() {
    	register("CoroCgoIterCallback", func() {
    		println("expect: OK")
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri May 17 19:46:10 UTC 2024
    - 3.4K bytes
    - Viewed (0)
Back to top