Searched refs:input_subscripts (Results 1 – 4 of 4) sorted by relevance
29 gtl::InlinedVector<string, 2>* input_subscripts, in ValidateEinsumEquation() argument38 *input_subscripts = in ValidateEinsumEquation()40 if (input_subscripts->size() != 1 && input_subscripts->size() != 2) { in ValidateEinsumEquation()43 "' but got: ", input_subscripts->size()); in ValidateEinsumEquation()
54 gtl::InlinedVector<string, 2>* input_subscripts,
424 input_subscripts = equation.split('->')[0].split(',')426 for subscript in input_subscripts:
1108 input_subscripts = equation.split('->')[0].split(',')1110 for subscript in input_subscripts: