Searched defs:input_labels (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | einsum.cc | 79 static bool OrderMatches(const Labels& input_labels, int offset, int m, in OrderMatches()
113 Status DetermineLayout(const Labels& input_labels, in DetermineLayout()
144 const EinsumLayout preferred_layout, const Labels& input_labels, in CalculateMixedLayoutPermutation()
190 Status Initialize(const TRT_TensorOrWeights& operand, Labels input_labels, in Initialize()
255 const TRT_TensorOrWeights& operand, Labels input_labels, in Create()
602 OperandLabels input_labels; in ParseEquation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | einsum_op_util.cc | 82 Status ParseEinsumEquation(const string& equation, OperandLabels* input_labels, in ParseEinsumEquation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_einsum_op.cc | 155 OperandLabels input_labels(mkl_input_labels_); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 161 gtl::InlinedVector<string, 2> input_labels; in EinsumShape() local
|