Searched refs:AddLoopsForShape (Results 1 – 5 of 5) sorted by relevance
234 IrArray::Index AddLoopsForShape(const Shape& shape, absl::string_view suffix);
234 IrArray::Index ForLoopNest::AddLoopsForShape(const Shape& shape, in AddLoopsForShape() function in xla::llvm_ir::ForLoopNest
712 source_loops.AddLoopsForShape(source->shape(), "source"); in HandleSelectAndScatter()734 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()2031 loops.AddLoopsForShape(operand->shape(), "operand"); in HandlePad()
2638 const IrArray::Index window_index = loops.AddLoopsForShape( in EmitElementalReduceWindow()
2295 const IrArray::Index window_index = window_loops.AddLoopsForShape( in EmitSelectAndScatterFromMlir()