Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetMlirOpNameFromOpCode (0.9 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_operator.h

    bool IsStablehloOp(const tflite::OperatorCodeT &op_code);
    
    // Returns the MLIR op name for the flatbuffer operator corresponding to
    // `op_code`.
    std::string GetMlirOpNameFromOpCode(const ::tflite::OperatorCodeT &op_code);
    
    // Returns the builtin op code for the given MLIR operation on success; emits
    // error and returns std::nullopt on failure.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 21:00:09 UTC 2024
    - 11.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

      llvm::StringRef op_name(
          tflite::EnumNameBuiltinOperator(tflite::GetBuiltinCode(&op_code)));
      return op_name.starts_with("STABLEHLO_");
    }
    
    std::string mlir::GetMlirOpNameFromOpCode(
        const tflite::OperatorCodeT& op_code) {
      auto builtin_code = tflite::GetBuiltinCode(&op_code);
      if (builtin_code == tflite::BuiltinOperator_IF) {
        return std::string("tf.If");
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

                              const tflite::OperatorCodeT& op_code) {
      if (IsBasicLSTMOp(op.builtin_options)) {
        return std::string("tfl.basic_lstm");
      }
      return mlir::GetMlirOpNameFromOpCode(op_code);
    }
    
    StatusOr<Operation*> BuildExternalConstOp(const tflite::TensorT& tensor,
                                              int32_t buffer_index,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top