Searched defs:reshape_b_only_on_first_run (Results 1 – 3 of 3) sorted by relevance
56 …signed int m, unsigned int n, unsigned int k, DataType data_type, bool reshape_b_only_on_first_run) in is_gemm_reshaped()
125 …signed int n, unsigned int k, unsigned int b, DataType data_type, bool reshape_b_only_on_first_run) in select_gemm_kernel()
2065 bool reshape_b_only_on_first_run() const in reshape_b_only_on_first_run() function