Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops.py769 for input_labels in input_axis_labels:
770 if (len(input_axis_labels) == 1 and input_labels.count(a) == 2 and
771 input_labels == input_labels[::-1] and '->' not in equation):
773 if input_labels.count(a) > 1:
776 input_labels)
1250 input_labels = match.group(1).split(',')
1253 if len(input_shapes) != len(input_labels):
1255 len(input_shapes), equation, len(input_labels)))
1289 for i, (labels, shape) in enumerate(zip(input_labels, input_shapes)):
1318 for labels in input_labels:
/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h127 OperandLabels* input_labels, in ParseEquation()
142 input_labels->resize(num_inputs); in ParseEquation()
146 MapToLabels(input_str[i], &input_labels->at(i), &label_mapping); in ParseEquation()
156 for (const int label : input_labels->at(i)) { in ParseEquation()
221 const bool output_has_ellipsis, OperandLabels* input_labels, in ProcessDimensions()
225 if (inputs.size() != input_labels->size()) { in ProcessDimensions()
226 return errors::InvalidArgument("Expected ", input_labels->size(), in ProcessDimensions()
237 Labels* labels = &(*input_labels)[i]; in ProcessDimensions()
605 OperandLabels input_labels(input_labels_); in Compute()
614 &input_labels, &output_labels, &label_types, in Compute()
[all …]
/external/autotest/utils/
Dlabellib_unittest.py70 input_labels = copy.deepcopy(labels)
71 mapping = labellib.LabelsMapping(input_labels)
73 self.assertEqual(input_labels, labels)
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc161 gtl::InlinedVector<string, 2> input_labels; in EinsumShape() local
164 ParseEinsumEquation(equation, &input_labels, &output_labels)); in EinsumShape()
170 const int input_labels_size = input_labels.size(); in EinsumShape()
172 return errors::InvalidArgument("Expected ", input_labels.size(), in EinsumShape()
183 TF_RETURN_IF_ERROR(ValidateEinsumEllipsis(input_labels[i], &has_ellipsis)); in EinsumShape()
190 static_cast<int>(input_labels[i].size()) - 3; in EinsumShape()
195 const int num_named_labels = static_cast<int>(input_labels[i].size()); in EinsumShape()
206 for (int label_idx = 0, end = input_labels[i].size(); label_idx < end; in EinsumShape()
208 const char label = input_labels[i][label_idx]; in EinsumShape()
215 ? label_idx + c->Rank(input_shape) - input_labels[i].size() in EinsumShape()