Home
last modified time | relevance | path

Searched refs:rhs_dimension_number (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc203 auto rhs_dimension_number = [&](int64 d) { in Einsum() local
210 dnums.add_rhs_batch_dimensions(rhs_dimension_number(dim_name)); in Einsum()
213 dnums.add_rhs_contracting_dimensions(rhs_dimension_number(dim_name)); in Einsum()