Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h202 LabelToDimSizes* label_to_dim_sizes) { in RecordLabelToDimension()
205 if (label_to_dim_sizes->at(label) != 0 && in RecordLabelToDimension()
206 label_to_dim_sizes->at(label) != input_dim) { in RecordLabelToDimension()
208 "Expected dimension ", label_to_dim_sizes->at(label), " at axis ", in RecordLabelToDimension()
212 (*label_to_dim_sizes)[label] = input_dim; in RecordLabelToDimension()
224 LabelToDimSizes* label_to_dim_sizes) { in ProcessDimensions()
235 label_to_dim_sizes->resize(num_named_labels); in ProcessDimensions()
248 label_to_dim_sizes)); in ProcessDimensions()
271 RecordLabelToDimension(label, axis, inputs[i], label_to_dim_sizes)); in ProcessDimensions()
610 LabelToDimSizes label_to_dim_sizes; in Compute() local
[all …]