Searched refs:dot_cols (Results 1 – 1 of 1) sorted by relevance
90 const int64 dot_cols = dot_shape.dimensions(num_batch_dims + 1); in DecomposeBatchDot() local91 Shape dot_shape_r2 = ShapeUtil::MakeShape(dot_type, {dot_rows, dot_cols}); in DecomposeBatchDot()92 Shape dot_shape_r3 = ShapeUtil::MakeShape(dot_type, {1, dot_rows, dot_cols}); in DecomposeBatchDot()94 ShapeUtil::MakeShape(dot_type, {batch_size, dot_rows, dot_cols}); in DecomposeBatchDot()