Searched refs:dot_rows (Results 1 – 1 of 1) sorted by relevance
89 const int64 dot_rows = dot_shape.dimensions(num_batch_dims + 0); 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()