Searched refs:fill_dims (Results 1 – 2 of 2) sorted by relevance
1241 auto fill_dims = [](mlir::DenseElementsAttr mlir_dim, auto* config_attrs) { in EmitGemmThunkFromMlir() local1245 fill_dims(mlir_dims.lhs_batching_dimensions(), in EmitGemmThunkFromMlir()1247 fill_dims(mlir_dims.rhs_batching_dimensions(), in EmitGemmThunkFromMlir()1249 fill_dims(mlir_dims.lhs_contracting_dimensions(), in EmitGemmThunkFromMlir()1251 fill_dims(mlir_dims.rhs_contracting_dimensions(), in EmitGemmThunkFromMlir()
787 fill_dims = orig.get_shape().as_list()[1:]802 [start_spatial_dim - 1 - prev_spatial_dim] + fill_dims,