in attention.h:
CostMetrics measure_operator_cost(SimEnvFactory const &sim,
MultiHeadAttentionAttrs const &attrs,
ParallelTensorShape const &query_shape,
ParallelTensorShape const &key_shape,
ParallelTensorShape const &value_shape,
ProfilingSettings const &settings,
MachineView const &mv);
in attention.cpp (correct):
CostMetrics measure_operator_cost(SimEnvFactory const &sim,
MultiHeadAttentionAttrs const &attrs,
InputParallelTensorDesc const &query_shape,
InputParallelTensorDesc const &key_shape,
InputParallelTensorDesc const &value_shape,
ProfilingSettings const &settings,
MachineView const &mv) {
in
attention.h:in
attention.cpp(correct):