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