Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for user_ops (0.29 sec)

  1. tensorflow/compiler/mlir/lite/tf_tfl_translate.cc

      toco_flags.set_reduce_type_precision(reduce_type_precision);
      // Read list of user select ops.
      llvm::SmallVector<llvm::StringRef, 2> user_ops;
      (llvm::StringRef(select_user_tf_ops))
          .split(user_ops, ',', /*MaxSplit=*/-1,
                 /*KeepEmpty=*/false);
      llvm::for_each(user_ops, [&toco_flags](llvm::StringRef op_name) {
        *(toco_flags.add_select_user_tf_ops()) = op_name.str();
      });
    
      std::string result;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 14K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc

        // Skip the original dequant op and connect the op before dequantize to the
        // user op.
        user_op->setOperand(user_idx, op_before_dequantize);
    
        // Wire input/output nodes.
        new_dequantize_op->setOperand(0, user_op->getResult(0));
        new_dequantize_op->getResult(0).setType(user_op->getResult(0).getType());
        user_op->getResult(0).replaceAllUsesExcept(new_dequantize_op->getResult(0),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc

        Operation* user_op, int argument_index, ElementsAttr value,
        llvm::SmallVector<std::pair<Region*, int>, 4>* work_list,
        llvm::MapVector<Operation*, llvm::SmallVector<unsigned int, 4>>*
            arguments_to_erase) {
      if (auto read_variable_op = dyn_cast<TF::ReadVariableOp>(user_op)) {
        (*arguments_to_erase)[read_variable_op];
        PropagateUsage(read_variable_op, value);
      } else if (auto call = dyn_cast<CallOpInterface>(user_op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 09:56:53 UTC 2024
    - 19.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc

        // together; epsilon (only with FusedBatchNorm).
        std::vector<NamedAttribute> attrs = contraction->getAttrs();
        ArrayAttr fused_ops_attr = ArrayAttr::get(context, fused_ops);
        attrs.push_back(
            NamedAttribute(StringAttr::get(context, "fused_ops"), fused_ops_attr));
        // Epsilon is used only in fusions with the FusedBatchNorm op, so we zero it
        // here.
        Attribute epsilon = rewriter.getF32FloatAttr(0);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    Performs a convolution followed by a specified series of operations.
      }];
    
      let description = [{
    The inputs to the convolution are `input` and `filter`. The series of operations
    that follows is specified by the `fused_ops` attribute, which is a list of TF op
    names specified as strings (e.g. "Relu"). They are performed in order, where the
    (first) input to each op is the output of the preceding op. The first input and
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top