Searched refs:array_multi_index (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | loop_emitter.cc | 108 std::vector<llvm::Value*> array_multi_index(shape_.dimensions_size()); in EmitStaticIndex() local
115 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitStaticIndex()
117 return IrArray::Index(array_multi_index, shape_, index_type); in EmitStaticIndex()
127 std::vector<llvm::Value*> array_multi_index(shape_.dimensions_size()); in EmitDynamicIndex() local
134 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitDynamicIndex()
136 return IrArray::Index(array_multi_index, shape_, index_type); in EmitDynamicIndex()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | parallel_loop_emitter.cc | 48 std::vector<llvm::Value*> array_multi_index(num_dims); in EmitIndexAndSetExitBasicBlock() local
63 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
70 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
80 llvm_ir::IrArray::Index array_index(array_multi_index, shape_, index_type); in EmitIndexAndSetExitBasicBlock()
|
| H A D | ir_emitter.cc | 1805 std::vector<llvm::Value*> array_multi_index( in EmitVectorizedReduce() local
1814 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1835 array_multi_index[innermost_dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1841 llvm_ir::IrArray::Index array_index(array_multi_index, reduce->shape(), in EmitVectorizedReduce()
1868 array_multi_index[innermost_dimension] = in EmitVectorizedReduce()
1875 llvm_ir::IrArray::Index array_index(array_multi_index, reduce->shape(), in EmitVectorizedReduce()
|