Searched refs:output_labels (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 128 Labels* output_labels, in ParseEquation()
148 MapToLabels(output_str, output_labels, &label_mapping); in ParseEquation()
164 for (const int label : *output_labels) { in ParseEquation()
222 Labels* output_labels, std::vector<DimensionType>* label_types, in ProcessDimensions()
287 std::find(output_labels->begin(), output_labels->end(), kEllipsisLabel); in ProcessDimensions()
288 if (it != output_labels->end()) { in ProcessDimensions()
289 const int ellipsis_axis = it - output_labels->begin(); in ProcessDimensions()
291 output_labels, output_label_counts); in ProcessDimensions()
606 Labels output_labels(output_labels_); in Compute()
614 &input_labels, &output_labels, &label_types, in Compute()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1251 output_labels = match.group(2)[2:] if match.group(2) else None
1261 output_labels = ''.join([
1265 resolved_equation += '->' + output_labels
1267 if output_labels and len(set(output_labels)) != len(output_labels):
1272 for label in output_labels:
1276 if ellipsis_label and output_labels:
1277 num_output_ellipses = output_labels.count(ellipsis_label)
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | generate_streaming_test_wav.py | 127 output_labels = []
159 output_labels.append({'label': wanted_label, 'time': output_offset_ms})
167 for output_label in output_labels:
|
| /external/tensorflow/tensorflow/lite/testing/model_coverage/ |
| D | model_coverage_lib.py | 323 output_labels = [output_tensor["name"] for output_tensor in output_details]
324 return output_data, output_labels
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 60 def _make_train_step_fn(model, mode, strategy, output_labels): argument
106 for label, output in zip(output_labels, combined_fn.outputs):
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 162 string output_labels; in EinsumShape() local
164 ParseEinsumEquation(equation, &input_labels, &output_labels)); in EinsumShape()
265 ValidateEinsumEllipsis(output_labels, &output_has_ellipsis)); in EinsumShape()
286 for (int label_idx = 0, end = output_labels.size(); label_idx < end; in EinsumShape()
288 const char label = output_labels[label_idx]; in EinsumShape()
|