Searched refs:input_multi_index (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1895 std::vector<llvm::Value*> input_multi_index(rank); in EmitElementalDynamicSlice() local
1899 input_multi_index[i] = Add(slice_start_multi_index[i], index[i]); in EmitElementalDynamicSlice()
1901 llvm_ir::IrArray::Index input_index(input_multi_index, input_hlo->shape(), in EmitElementalDynamicSlice()
2644 std::vector<llvm::Value*> input_multi_index(index.size()); in EmitElementalReduceWindow() local
2649 input_multi_index[i] = NSWSub( in EmitElementalReduceWindow()
2658 ICmpEQ(SRem(input_multi_index[i], in EmitElementalReduceWindow()
2664 input_multi_index[i] = in EmitElementalReduceWindow()
2665 SDiv(input_multi_index[i], in EmitElementalReduceWindow()
2675 ICmpULT(input_multi_index[i], in EmitElementalReduceWindow()
2687 input_multi_index, reduce_window->input_arrays()[0]->shape(), index_type); in EmitElementalReduceWindow()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1560 std::vector<llvm::Value*> input_multi_index = in EmitInnerLoopForVectorizedReduction() local
1569 for (auto& i : input_multi_index) { in EmitInnerLoopForVectorizedReduction()
1575 llvm_ir::IrArray::Index input_index(input_multi_index, arg->shape(), in EmitInnerLoopForVectorizedReduction()
|