Searched refs:GetReductionToVectorDimensions (Results 1 – 1 of 1) sorted by relevance
3524 std::tuple<int64, int64, int64> GetReductionToVectorDimensions( in GetReductionToVectorDimensions() function3665 GetReductionToVectorDimensions(input_shape, first_reduce->dimensions()); in ComputeMappingSchemeAndReductionKind()