Searched refs:concat_dim_size (Results 1 – 2 of 2) sorted by relevance
499 int64* concat_dim_size) { in FindMklCommonFormat() argument501 *concat_dim_size = 0; in FindMklCommonFormat()508 *concat_dim_size += src_dims[concat_dim]; in FindMklCommonFormat()
1792 auto concat_dim_size = source_index.GetConstantWithIndexType( in EmitElementalConcatenate() local1797 CondBr(ICmpULT(source_multi_index[concat_dim], concat_dim_size), in EmitElementalConcatenate()1804 Sub(source_multi_index[concat_dim], concat_dim_size); in EmitElementalConcatenate()