Searched refs:input_multi_index (Results 1 – 2 of 2) sorted by relevance
2086 std::vector<llvm::Value*> input_multi_index(rank); in EmitElementalDynamicSlice() local2090 input_multi_index[i] = Add(slice_start_multi_index[i], index[i]); in EmitElementalDynamicSlice()2092 llvm_ir::IrArray::Index input_index(input_multi_index, input_hlo->shape(), in EmitElementalDynamicSlice()2841 std::vector<llvm::Value*> input_multi_index(index.size()); in EmitElementalReduceWindow() local2846 input_multi_index[i] = NSWSub( in EmitElementalReduceWindow()2855 ICmpEQ(SRem(input_multi_index[i], in EmitElementalReduceWindow()2861 input_multi_index[i] = in EmitElementalReduceWindow()2862 SDiv(input_multi_index[i], in EmitElementalReduceWindow()2872 ICmpULT(input_multi_index[i], in EmitElementalReduceWindow()2883 IrArray::Index input_index(input_multi_index, in EmitElementalReduceWindow()[all …]
1658 std::vector<llvm::Value*> input_multi_index = in EmitInnerLoopForVectorizedReduction() local1667 for (auto& i : input_multi_index) { in EmitInnerLoopForVectorizedReduction()1673 llvm_ir::IrArray::Index input_index(input_multi_index, arg->shape(), in EmitInnerLoopForVectorizedReduction()