Searched refs:AddLoopsForShape (Results 1 – 5 of 5) sorted by relevance
232 IrArray::Index AddLoopsForShape(const Shape& shape, absl::string_view suffix);
235 IrArray::Index ForLoopNest::AddLoopsForShape(const Shape& shape, in AddLoopsForShape() function in xla::llvm_ir::ForLoopNest
735 source_loops.AddLoopsForShape(source->shape(), "source"); in HandleSelectAndScatter()758 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()2142 loops.AddLoopsForShape(operand->shape(), "operand"); in HandlePad()
2835 const IrArray::Index window_index = loops.AddLoopsForShape( in EmitElementalReduceWindow()
2007 const IrArray::Index window_index = window_loops.AddLoopsForShape( in EmitSelectAndScatter()