Searched defs:CollapseFirstNDims (Results 1 – 2 of 2) sorted by relevance
869 Shape CollapseFirstNDims(const Shape& shape, int64 n) { in CollapseFirstNDims() function882 llvm_ir::IrArray CollapseFirstNDims(llvm::IRBuilder<>* b, in CollapseFirstNDims() function
311 StatusOr<HloInstruction*> CollapseFirstNDims(HloInstruction* operand, int64 n) { in CollapseFirstNDims() function