Searched refs:DotOperationType (Results 1 – 1 of 1) sorted by relevance
1071 enum class DotOperationType { enum1078 DotOperationType GetDotOperationType(mhlo::DotOp dot_op) { in GetDotOperationType()1089 return DotOperationType::kVectorDot; in GetDotOperationType()1093 return DotOperationType::kMatrixVector; in GetDotOperationType()1097 return DotOperationType::kMatrixMatrix; in GetDotOperationType()1099 return DotOperationType::kUnsupported; in GetDotOperationType()1104 DotOperationType type) { in GetDotOpInitTensorDynSizes()1107 case DotOperationType::kMatrixMatrix: { in GetDotOpInitTensorDynSizes()1114 case DotOperationType::kMatrixVector: { in GetDotOpInitTensorDynSizes()1119 case DotOperationType::kVectorDot: in GetDotOpInitTensorDynSizes()[all …]