- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for kOpKeyAttrName (0.29 sec)
-
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
#include "tensorflow/core/tfrt/fallback/cost_recorder.h" #include "tfrt/init_tfrt_dialects.h" // from @tf_runtime 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) {
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/constants.h
#define TENSORFLOW_COMPILER_MLIR_TFRT_CONSTANTS_H_ namespace tensorflow { namespace tfrt_compiler { // Use __ prefix to indicate this is internal attribute. inline constexpr char kOpKeyAttrName[] = "__op_key"; } // namespace tfrt_compiler } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 09 04:34:14 UTC 2023 - 1K bytes - Viewed (0) -
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)