Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Deinsum_op_util.cc28 gtl::InlinedVector<string, 2>* input_subscripts, in ParseEinsumEquation() argument
37 *input_subscripts = in ParseEinsumEquation()
39 if (input_subscripts->size() != 1 && input_subscripts->size() != 2) { in ParseEinsumEquation()
42 "' but got: ", input_subscripts->size()); in ParseEinsumEquation()
Deinsum_op_util.h25 gtl::InlinedVector<string, 2>* input_subscripts,
/external/tensorflow/tensorflow/python/kernel_tests/
Deinsum_op_test.py428 input_subscripts = equation.split('->')[0].split(',')
430 for subscript in input_subscripts:
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops_test.py1114 input_subscripts = equation.split('->')[0].split(',')
1116 for subscript in input_subscripts: