Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 96 for op_names (0.39 sec)

  1. tensorflow/c/eager/tracing_utils.cc

    namespace tracing {
    
    Status MaybeSetOpName(AbstractOperation* op, const char* op_name) {
      if (isa<TracingOperation>(op)) {
        TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name));
      }
      if (isa<gradients::TapeOperation>(op)) {
        TF_RETURN_IF_ERROR(MaybeSetOpName(
            dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(),
            op_name));
      }
      return absl::OkStatus();
    }
    }  // namespace tracing
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 13:57:45 UTC 2024
    - 1.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/tftext_utils_test.cc

    using tensorflow::OpRegistrationData;
    using tensorflow::OpRegistry;
    using tensorflow::Status;
    
    namespace {
    
    void Register(const std::string& op_name, OpRegistry* registry) {
      registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status {
        op_reg_data->op_def.set_name(op_name);
        return absl::OkStatus();
      });
    }
    
    }  // namespace
    
    TEST(TfTextUtilsTest, TestTfTextRegistered) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:14:37 UTC 2024
    - 1.7K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc

    }
    
    }  // namespace
    
    void RegisterCostFunction(absl::string_view op_name,
                              CostFunction cost_function) {
      RegisterCostFunction(GetCostFunctionRegistry(), op_name,
                           std::move(cost_function));
    }
    
    bool HasCostFunctionRegistered(absl::string_view op_name) {
      return GetCostFunctionRegistry().contains(op_name);
    }
    
    int64_t CostAnalysis::GetCost(mlir::Operation* op) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc

        "/tensorflow/core/tf2xla/v1/mlir_failed_xla_legalize_tf_count",
        "Counts the attempts of legalization of ops", "op_name");
    
    auto *mlir_failed_legalization_count = tensorflow::monitoring::Counter<2>::New(
        "/tensorflow/core/tf2xla/v1/mlir_failed_xla_legalize_tf_pass_count",
        "Counts the failure of legalization of ops", "op_name", "legality");
    
    class LegalizeTF : public impl::LegalizeTFBase<LegalizeTF> {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 10.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc

      bool IsAllowListedOp(Operation *op) {
        std::string op_name = op->getName().getStringRef().str();
        if (IsAlwaysAllowlistedOp(op) || IsTfFakeQuantOp(op)) {
          return true;
        } else if (mode_ == kDefaultMode) {
          return QuantizableOpsInDefaultMode().count(op_name) > 0;
        } else if (mode_ == kLegacyIntegerMode) {
          return QuantizableOpsInLegacyMode().count(op_name) > 0;
        } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  8. 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)
  9. 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)
  10. 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)
Back to top