Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 149 for op_names (0.32 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc

        dialect_name_of_[op_with_dialect_name] = dialect_name;
        op_name_of_[op_with_dialect_name] = op_name;
        ++total_ops_;
      });
      PrintSummary();
    }
    
    void PrintOpStatsPass::CountOp(
        DenseMap<StringRef, llvm::StringMap<int64_t>> &op_count_map,
        StringRef op_name, StringRef dtype) {
      auto &op_counts = op_count_map[op_name];
      if (auto it = op_counts.find(dtype); it != op_counts.end()) {
        it->second++;
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization.cc

            "mlir_second_phase_failed_legalization_op_count",
            "Counts which op fails to legalize", "op_name");
    
    auto* mlir_non_static_op_count = tensorflow::monitoring::Counter<1>::New(
        "/tensorflow/core/tf2xla/"
        "mlir_second_phase_non_static_op_count",
        "Counts which ops do not have static results", "op_name");
    
    auto* mlir_non_static_op_skip_count = tensorflow::monitoring::Counter<1>::New(
        "/tensorflow/core/tf2xla/"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc

          }
        }
    
        return success();
      }
    
     private:
      // Parses TFLite Flex op's `custom_options` and returns the TF
      // `op_name` and TF dialect MLIR op attributes.
      static LogicalResult ParseCustomOption(
          StringRef custom_options, Location loc, std::string& op_name,
          SmallVectorImpl<NamedAttribute>& attributes,
          tensorflow::NodeDef& node_def) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.h

    };
    
    struct CostContext {
      int64_t default_unranked_tensor_size;
    };
    
    using CostFunction =
        std::function<int64_t(const CostContext&, mlir::Operation*)>;
    
    void RegisterCostFunction(absl::string_view op_name,
                              CostFunction cost_function);
    
    template <typename OpType, typename F>
    void RegisterCostFunction(F f) {
      RegisterCostFunction(
          OpType::getOperationName().str(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_async.cc

      CreateOp op = *this;
      p << "(" << op.getInCh() << ") key("
        << op->getAttrOfType<mlir::IntegerAttr>("op_key").getInt() << ") device("
        << op->getAttr("device") << ") " << op->getAttr("op_name") << "()";
    
      fallback_common::PrintExecuteOpCommon(p, op);
      fallback_common::PrintExecuteOpFuncAttribute(p, op);
    
      p << " num_args(" << op->getAttrOfType<mlir::IntegerAttr>("num_args").getInt()
        << ')';
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 15.6K bytes
    - Viewed (0)
Back to top