Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for function_ref (0.18 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    /// `intCalculate` is chosen to conduct the calculate.
    Attribute ConstFoldBinaryOp(
        Type result_type, ArrayRef<Attribute> operands,
        llvm::function_ref<APFloat(APFloat, APFloat)> float_calculate,
        llvm::function_ref<APInt(APInt, APInt)> int_calculate) {
      // Note: All types are wrapped in tensor types in TFlite. E.g., f32 is
      // represented as tensor<f32>. So we are only handling tensor types here.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

      return success();
    }
    
    static LogicalResult VerifyCaseOrIfOpBranchFunctions(
        SymbolTableCollection& symbol_table, Operation* op,
        ArrayRef<Attribute> branches,
        llvm::function_ref<std::string(unsigned branch_index)> branch_name) {
      SmallVector<FunctionType, 2> branch_types;
      branch_types.reserve(branches.size());
    
      if (llvm::any_of(op->getOperands(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      }
    };
    
    using ValuePortResultMap =
        absl::flat_hash_map<ValuePort, Attribute, ValuePortHasher>;
    using ComputedQueryFn = function_ref<bool(ValuePort)>;
    using ValueQueryFn = function_ref<Attribute(const ValuePort&)>;
    using ValuePortInputs = SmallVectorImpl<ValuePort>;
    
    // Note: Following implements the rank 1 pack op case so could be
    // generalized.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

                                  n->DebugString());
        }
      }
      return absl::OkStatus();
    }
    
    // Cheap check to tell whether FunctionDef contains a lifted argument.
    bool HasLiftedArgs(const FunctionDef& function_def) {
      return absl::c_any_of(function_def.node_def(), [](const NodeDef& node_def) {
        return (node_def.op() == "Placeholder" &&
                node_def.attr().find(kXlaLiftedArgOutsideCompilationAttrName) !=
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //===----------------------------------------------------------------------===//
    // ReshapeOp
    //===----------------------------------------------------------------------===//
    
    namespace {
    using ReshapeErrorHandler =
        llvm::function_ref<LogicalResult(const llvm::Twine &)>;
    
    LogicalResult GetReshapeOutputType(Value tensor, Value shape,
                                       ReshapeErrorHandler error_handler,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

                                 const FunctionDefLibrary& actual, string* diff) {
      std::unordered_map<string, const FunctionDef*> actual_index;
      for (const FunctionDef& function : actual.function()) {
        actual_index[function.signature().name()] = &function;
      }
    
      for (const FunctionDef& expected_function : expected.function()) {
        auto it = actual_index.find(expected_function.signature().name());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

    }
    
    Status ImporterBase::ConvertDeferredFunctions() {
      while (!deferred_functions_.empty()) {
        auto conversion_metadata = deferred_functions_.front();
        deferred_functions_.pop();
    
        const FunctionDef* func_def =
            graph_flib_.Find(conversion_metadata.function_name);
        // Converts the graph to an MLIR function and adds it to the module.
        // We populate the NodeSpec so that all the _Arg ops get their shape
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top