Searched refs:AddLoopsForShape (Results 1 – 5 of 5) sorted by relevance
234 IrArray::Index AddLoopsForShape(const Shape& shape, absl::string_view suffix);
233 IrArray::Index ForLoopNest::AddLoopsForShape(const Shape& shape, in AddLoopsForShape() function in xla::llvm_ir::ForLoopNest
382 const IrArray::Index window_index = loops.AddLoopsForShape( in MakeElementGenerator()
858 const IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()
662 const llvm_ir::IrArray::Index window_index = loops.AddLoopsForShape( in EmitElementalReduceWindow()793 source_loops.AddLoopsForShape(source->shape(), "source"); in HandleSelectAndScatter()815 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter()2117 loops.AddLoopsForShape(operand->shape(), "operand"); in HandlePad()