Searched defs:EinsumHelper (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | einsum_op_impl.h | 65 struct EinsumHelper { struct
69 enum DimensionType {
92 static DimensionType GetDimensionType(bool is_removed, bool is_unique) { in GetDimensionType()
104 static void MapToLabels(const string& subscript, Labels* labels, in MapToLabels()
126 static Status ParseEquation(const string& equation, in ParseEquation()
184 static void InsertBroadcastLabels(int num_bcast_dims, int num_named_labels, in InsertBroadcastLabels()
200 static Status RecordLabelToDimension(const int label, const int axis, in RecordLabelToDimension()
218 static Status ProcessDimensions( in ProcessDimensions()
304 static void PermuteLabels(const std::vector<int>& permutation, in PermuteLabels()
314 static Status CopyFrom(const Tensor& input, const TensorShape& shape, in CopyFrom()
[all …]
|