- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetCostForFunc (0.13 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.h
// Get the estimated cost for the op under the given hardware spec senario. float GetCostForOp(Operation* op, const std::string& hardware); // Get the estimated cost for the whole function under the given hardware. float GetCostForFunc(func::FuncOp* func, const std::string& hardware); // Get the transfer cost given from & to hardware info. // We will only calculate for the "necessary" tensor transferred.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/compute_cost.cc
auto target = GetTargetAnnotation(func); if (!target.has_value()) { func.emitError("we cannot get hardware info for this function."); signalPassFailure(); } float total_cost = GetCostForFunc(&func, *target); OpBuilder builder(func); UpdateCost(func, total_cost, &builder); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateComputeCostPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 24 15:10:02 UTC 2022 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc
auto* device_hardware = GetTargetHardware(hardware); if (device_hardware == nullptr) { return kDefaultFixedValuedCost; } return device_hardware->GetOpCost(op); } float GetCostForFunc(func::FuncOp* func, const std::string& hardware) { auto* device_hardware = GetTargetHardware(hardware); if (device_hardware == nullptr) { return kDefaultFixedValuedCost; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0)