Searched defs:LinalgCheckMatrixUnaryRuleHelper (Results 1 – 1 of 1) sorted by relevance
178 struct LinalgCheckMatrixUnaryRuleHelper<op_name, F, Func, typelist<A, T...>> { struct179 …tic inline Tensor check_and_reshape_input(const Tensor& tensor, std::optional<int64_t> batch_dim) { in check_and_reshape_input()184 static oneOutput apply_one( in apply_one()192 static twoOutputs apply_two( in apply_two()201 static threeOutputs apply_three( in apply_three()210 static fourOutputs apply_four( in apply_four()