Searched refs:ValidateDotDimensionNumbers (Results 1 – 2 of 2) sorted by relevance
896 Status ValidateDotDimensionNumbers(const DotDimensionNumbers& dim_numbers) { in ValidateDotDimensionNumbers() function934 TF_RETURN_IF_ERROR(ValidateDotDimensionNumbers(dot.dot_dimension_numbers())); in EmitBatchDotOperation()
557 Status ValidateDotDimensionNumbers( in ValidateDotDimensionNumbers() function634 TF_RETURN_IF_ERROR(ValidateDotDimensionNumbers(lhs, rhs, dimension_numbers)); in InferDotOpShape()