- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for op_count_map (0.18 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc
op_with_dialect_count_[op_with_dialect_name]); } auto op_count_map = op_dtype_count_[op_with_dialect_name]; if (!op_count_map.empty()) { std::string delim; *os_ << " ("; for (const auto &[dtype, cnt] : op_count_map) { *os_ << delim << absl::StrFormat("%s: %d", dtype, cnt); delim = ", "; } *os_ << ")"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
absl::flat_hash_map<int64_t, uint64_t> op_cost_map; op.walk([&](mlir::Operation* op) { const auto cost_attr = op->getAttrOfType<mlir::IntegerAttr>(kCostAttrName); if (!cost_attr) return; const auto op_key_attr = op->getAttrOfType<mlir::IntegerAttr>(kOpKeyAttrName); if (!op_key_attr) return; op_cost_map[op_key_attr.getInt()] = cost_attr.getInt(); }); return op_cost_map; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.h
#include "mlir/IR/BuiltinOps.h" // from @llvm-project #include "tensorflow/core/platform/status.h" #include "tensorflow/core/tfrt/fallback/cost_recorder.h" #include "tensorflow/core/tfrt/fallback/op_cost_map.pb.h" namespace tensorflow { namespace tfrt_compiler { // Analyze costs for tensorflow operations. // // The current heuristic used is quite simple, which is to calculate the total
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0)