Skip to content

Commit 18b1bae

Browse files
Automated Code Change
PiperOrigin-RevId: 828305086
1 parent 2d8bcd3 commit 18b1bae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

xla/backends/cpu/ynn_emitter.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,7 @@ static absl::StatusOr<YnnSubgraph> EmitYnnDotSubgraph(
431431
TF_ASSIGN_OR_RETURN(DotCanonicalDims dot_canonical_dims,
432432
GetDotCanonicalDims(dot_dimensions, dot_shape));
433433

434-
const size_t b_rank = rhs_shape.dimensions_size();
434+
const size_t b_rank = rhs_shape.dimensions().size();
435435
const bool transpose_b = !dot_canonical_dims.rhs_canonical;
436436
YNN_RETURN_IF_ERROR(DefineBatchMatrixMultiply(subgraph.get(), lhs_id, rhs_id,
437437
out_id, b_rank, transpose_b));

0 commit comments

Comments
 (0)