Searched defs:output_labels (Results 1 – 6 of 6) sorted by relevance
60 OutputContainer& output_labels) { in ConvertAndFillLabels()113 OutputContainer& output_labels) { in SelectConvertAndFillLabels()
83 Labels* output_labels, in ParseEinsumEquation()
156 Labels output_labels(mkl_output_labels_); in Compute() local
565 Labels output_labels) { in GetOutputTranspose()603 Labels output_labels; in ParseEquation() local
55 def _make_train_step_fn(model, mode, strategy, output_labels): argument
162 string output_labels; in EinsumShape() local