Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getDeviceName (0.36 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc

    }
    
    bool HasValidHardwareTarget(mlir::Operation* op) {
      // All TFLite ops has CPU interface, should be enough to check for cpu.
      return IsOpSupported(op, "CPU");
    }
    
    std::optional<std::string> GetDeviceName(mlir::Operation* op) {
      if (IsConst(op)) return std::nullopt;
    
      // The model may contain quant stats op which is unrelevant to the
      // execution.
      if (llvm::isa<mlir::func::ReturnOp, mlir::quantfork::StatisticsOp>(op))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 06:11:34 UTC 2024
    - 7.5K bytes
    - Viewed (0)
Back to top