Enzyme main
Loading...
Searching...
No Matches
mlir::enzyme::batchutils::BatchCacheKey Struct Reference

#include "MLIR/Passes/EnzymeBatchPass.h"

Collaboration diagram for mlir::enzyme::batchutils::BatchCacheKey:

Public Member Functions

bool operator< (const BatchCacheKey &other) const
 

Public Attributes

FunctionOpInterface function
 
SmallVector< int64_t > batchSizes
 

Detailed Description

Definition at line 17 of file EnzymeBatchPass.h.

Member Function Documentation

◆ operator<()

bool mlir::enzyme::batchutils::BatchCacheKey::operator< ( const BatchCacheKey & other) const
inline

Definition at line 22 of file EnzymeBatchPass.h.

References batchSizes, and function.

Member Data Documentation

◆ batchSizes

SmallVector<int64_t> mlir::enzyme::batchutils::BatchCacheKey::batchSizes

Definition at line 19 of file EnzymeBatchPass.h.

Referenced by operator<().

◆ function

FunctionOpInterface mlir::enzyme::batchutils::BatchCacheKey::function

Definition at line 18 of file EnzymeBatchPass.h.

Referenced by operator<().


The documentation for this struct was generated from the following file: