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