Searched refs:output_labels (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 102 Labels* output_labels, std::vector<EinsumDimensionType>* label_types, in ProcessDimensions()
167 std::find(output_labels->begin(), output_labels->end(), kEllipsisLabel); in ProcessDimensions()
168 if (it != output_labels->end()) { in ProcessDimensions()
169 const int ellipsis_axis = it - output_labels->begin(); in ProcessDimensions()
171 output_labels, output_label_counts); in ProcessDimensions()
492 Labels output_labels(output_labels_); in Compute()
500 &input_labels, &output_labels, &label_types, in Compute()
584 std::vector<int> output_permutation(output_labels.size()); in Compute()
592 for (int i = 0; i < output_labels.size(); ++i) { in Compute()
593 output_permutation[i] = label_to_position[output_labels[i]]; in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_einsum_op.cc | 156 Labels output_labels(mkl_output_labels_); in Compute() local
165 &output_labels, &label_types, &input_label_counts, in Compute()
245 std::vector<int> output_permutation(output_labels.size()); in Compute()
253 for (int i = 0; i < output_labels.size(); ++i) { in Compute()
254 output_permutation[i] = label_to_position[output_labels[i]]; in Compute()
256 label_to_position[output_labels[i]] += 1; in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | einsum_op_util.cc | 83 Labels* output_labels, in ParseEinsumEquation() argument
102 MapToLabels(output_str, output_labels, &label_mapping); in ParseEinsumEquation()
120 for (const int label : *output_labels) { in ParseEinsumEquation()
|
| D | einsum_op_util.h | 62 Labels* output_labels,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | special_math_ops.py | 1270 output_labels = match.group(2)[2:] if match.group(2) else None
1280 output_labels = ''.join([
1284 resolved_equation += '->' + output_labels
1286 if output_labels and len(set(output_labels)) != len(output_labels):
1291 for label in output_labels:
1295 if ellipsis_label and output_labels:
1296 num_output_ellipses = output_labels.count(ellipsis_label)
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | generate_streaming_test_wav.py | 123 output_labels = []
155 output_labels.append({'label': wanted_label, 'time': output_offset_ms})
163 for output_label in output_labels:
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | einsum.cc | 565 Labels output_labels) { in GetOutputTranspose() argument
575 absl::MakeConstSpan(output_labels.begin(), in GetOutputTranspose()
576 output_labels.end()), in GetOutputTranspose()
603 Labels output_labels; in ParseEquation() local
610 ParseEinsumEquation(equation, &input_labels, &output_labels, &label_types, in ParseEquation()
658 GetOutputTranspose(**descriptor_a, **descriptor_b, output_labels); in ParseEquation()
818 Labels output_labels;
826 eq, &input_labels, &output_labels, &label_types, &input_label_counts,
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-192.md | 22 for (const int label : *output_labels) {
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 55 def _make_train_step_fn(model, mode, strategy, output_labels): argument
101 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 ValidateEinsumEquation(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()
|