Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetCostForOp (0.23 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.cc

          auto hardware = GetTargetAnnotation(op);
          if (!hardware) return;
          float cost = GetCostForOp(op, *hardware);
          UpdateCost(op, cost, &builder);
        }
      });
    }
    
    }  // namespace
    
    float GetCostForOp(Operation* op, const std::string& hardware) {
      auto* device_hardware = GetTargetHardware(hardware);
      if (device_hardware == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/experimental/tac/transforms/cost_model.h

    // analyze the cost.
    
    // ================== Interface  ========================
    
    // 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);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 19 00:13:50 UTC 2022
    - 2.5K bytes
    - Viewed (0)
Back to top