Searched refs:window_index (Results 1 – 4 of 4) sorted by relevance
2844 DimensionVector window_index(rank);2845 std::fill(window_index.begin(), window_index.end(), 0);2865 window_index[i] * window.dimensions(i).window_dilation() -2881 IndexUtil::BumpIndices(window_shape, absl::MakeSpan(window_index)));
2638 const IrArray::Index window_index = loops.AddLoopsForShape( in EmitElementalReduceWindow() local2640 CHECK_EQ(window_index.size(), index.size()); in EmitElementalReduceWindow()2652 NSWMul(window_index[i], in EmitElementalReduceWindow()
734 const llvm_ir::IrArray::Index window_index = window_loops.AddLoopsForShape( in HandleSelectAndScatter() local747 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
2295 const IrArray::Index window_index = window_loops.AddLoopsForShape( in EmitSelectAndScatterFromMlir() local2318 operand_multi_index[i] = NSWSub(NSWAdd(strided_index, window_index[i]), in EmitSelectAndScatterFromMlir()