Searched refs:concat_dim_size (Results 1 – 3 of 3) sorted by relevance
872 int64* concat_dim_size) { in FindMklCommonFormat() argument874 *concat_dim_size = 0; in FindMklCommonFormat()881 *concat_dim_size += src_dims[concat_dim]; in FindMklCommonFormat()
887 int64 concat_dim_size = 0; in ExchangeHalo() local889 concat_dim_size += piece->shape().dimensions(dim); in ExchangeHalo()891 concat_shape.set_dimensions(dim, concat_dim_size); in ExchangeHalo()
1834 int64 concat_dim_size = 0; in EmitElementalConcatenate() local1842 source_index.GetConstantWithIndexType(concat_dim_size), in EmitElementalConcatenate()1844 concat_dim_size += operand->shape().dimensions(concat_dim); in EmitElementalConcatenate()1846 source_index.GetConstantWithIndexType(concat_dim_size)), in EmitElementalConcatenate()