Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 157 for op_names (0.17 sec)

  1. tensorflow/compiler/mlir/tfr/python/tfr_gen.py

      def _visit_tfr_builtins(self, op_name, args, node):
        arg_strs = []
        arg_tys = []
        for arg in args:
          value, ty = self.visit(arg)
          arg_strs.append(value)
          arg_tys.append(ty)
        tfr_op_name = 'tfr.' + op_name[5:]
        ret_tys = (
            TFR_BUILTINS[op_name](*arg_tys)
            if callable(TFR_BUILTINS[op_name]) else TFR_BUILTINS[op_name])
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 27 15:27:03 UTC 2022
    - 55.8K bytes
    - Viewed (0)
  2. tensorflow/c/eager/gradients.cc

    }
    }  // namespace
    
    Status GradientRegistry::Register(
        const string& op_name, GradientFunctionFactory gradient_function_factory) {
      auto iter = registry_.find(op_name);
      if (iter != registry_.end()) {
        const string error_msg = "Gradient already exists for op: " + op_name + ".";
        return errors::AlreadyExists(error_msg);
      }
      registry_.insert({op_name, gradient_function_factory});
      return absl::OkStatus();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 15 09:49:45 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py

    
    def _find_op(
        graph: ops.Graph, op_name: Optional[str]
    ) -> Optional[ops.Operation]:
      """Finds the operation with `op_name`.
    
      Args:
        graph: The graph to find from.
        op_name: Name of the node.
    
      Returns:
        The operation that corresponds to `op_name`. Returns None iff op_name is an
        empty string or None.
    
      Raises:
        ValueError: `op_name` is malformed.
      """
      if not op_name:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/metrics/types_util.cc

        const std::string& op_name, const Location& location) {
      using tflite::metrics::ConverterErrorData;
      ConverterErrorData error;
      if (!pass_name.empty()) {
        error.set_subcomponent(pass_name);
      }
    
      if (!error_message.empty()) {
        error.set_error_message(error_message);
      }
    
      if (!op_name.empty()) {
        error.mutable_operator_()->set_name(op_name);
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/quantization/quantization_info.proto

      }
    
      // The quantization parameters for a named tensor.
      message QuantParams {
        // The tensor name has the following convention:
        //     tensor_name := op_name | op_name  ’:’  port_number.
        // If the op has only one port, op_name can be used.
        // If the op has internal states, such as fused LSTM, the port_number should
        // follow a predefined convention.
        oneof name_oneof {
          string name = 1;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 08 03:45:04 UTC 2019
    - 2.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

    absl::StatusOr<bool> IsSmallConstant(Node* n) {
      if (!n->IsConstant()) {
        return false;
      }
    
      return IsConstantSmall(n);
    }
    
    bool IsInPlaceOp(absl::string_view op_name) {
      return op_name == "InplaceUpdate" || op_name == "InplaceAdd" ||
             op_name == "InplaceSub";
    }
    }  // namespace
    
    Status CloneConstantsForBetterClusteringPassImpl::CloneSmallConstantInputs(
        const absl::flat_hash_set<string>& name_set, Node* n) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/ops/gen/README.md

    # TensorFlow Op CodeGen Machinery (Experimental)
    
    ## Usage
    
    ```
    usage: generate_cpp  [flags]  OpName1 [OpName2 ...]
    Flags:
        --help=false                        bool    Print this help message.
        --category=""                       string  Category for generated ops (e.g. 'math', 'array').
        --namespace=""                      string  Compact C++ namespace, default is 'tensorflow::ops'.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 21 18:51:25 UTC 2021
    - 5.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          return true;
        }
      }
      return false;
    }
    
    // Returns whether the current op is not supported by the TF Lite runtime.
    static bool IsUnsupportedFlexOp(const std::string& op_name) {
      return op_name == "PartitionedCall" || op_name == "StatefulPartitionedCall";
    }
    
    // Create description of operation that could not be converted.
    static std::string GetOpDescriptionForDebug(Operation* inst) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc

           parser.parseAttribute(device, "device", result.attributes) ||
           parser.parseRParen()))
        return mlir::failure();
    
      if (options.has_op_name &&
          parser.parseAttribute(op_name, "op_name", result.attributes))
        return mlir::failure();
    
      if (options.has_symbol_ref &&
          parser.parseAttribute(f, "f", result.attributes))
        return mlir::failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  10. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc

        return "";
      }
      return std::string(entry->second);
    }
    
    string WriteClassFuzzDef(const OpInfo& op_info) {
      string class_signature_str = absl::Substitute(
          "class Fuzz$0 : public FuzzSession<$1> {\n", op_info.op_name,
          absl::StrJoin(op_info.graph_op_def.input_arg(), ", ",
                        [](string* out, const auto arg) {
                          strings::StrAppend(out, "Tensor");
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 27 16:26:51 UTC 2024
    - 13K bytes
    - Viewed (0)
Back to top