- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for kCostAttrName (0.21 sec)
-
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
namespace tensorflow { namespace { constexpr char kCostAttrName[] = "_tfrt_cost"; constexpr char kOpKeyAttrName[] = "op_key"; absl::flat_hash_map<int64_t, uint64_t> GetOpCostMap(mlir::ModuleOp op) { 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 =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0)