- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for kOpKeyAttrName (0.09 sec)
-
tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc
return; } // Try to use the recorded cost if any. if (cost_recorder_ != nullptr) { const auto op_key_attr = op->getAttrOfType<mlir::IntegerAttr>(kOpKeyAttrName); if (op_key_attr) { cost_map_[op] = cost_recorder_->GetCost(op_key_attr.getInt()); return; } } // These ops are cheap regardless of their input sizes. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0)