Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Deinsum_op_util.h25 using LabelCounts = gtl::InlinedVector<int, 8>; variable
26 using OperandLabelCounts = gtl::InlinedVector<LabelCounts, 2>;
65 LabelCounts* output_label_counts,
Deinsum_op_util.cc86 LabelCounts* output_label_counts, in ParseEinsumEquation()
/external/tensorflow/tensorflow/core/kernels/linalg/
Deinsum_op_impl.h58 using LabelCounts = gtl::InlinedVector<int, 8>; variable
59 using OperandLabelCounts = gtl::InlinedVector<LabelCounts, 2>;
66 LabelCounts* label_counts) { in InsertBroadcastLabels()
103 OperandLabelCounts* input_label_counts, LabelCounts* output_label_counts, in ProcessDimensions()
244 const LabelCounts& label_counts, in StrideOrInflate()
339 const LabelCounts& label_counts, Labels* labels, Labels* free_labels, in ReduceOperand()
495 LabelCounts output_label_counts(output_label_counts_); in Compute()
622 LabelCounts output_label_counts_;
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_einsum_op.cc159 LabelCounts output_label_counts(mkl_output_label_counts_); in Compute()
270 LabelCounts mkl_output_label_counts_;
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Deinsum.cc606 LabelCounts output_label_counts; in ParseEquation()
630 auto no_duplicated_labels = [](const LabelCounts& label_counts) { in ParseEquation()
821 LabelCounts output_label_counts;