Home
last modified time | relevance | path

Searched refs:array_multi_index (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc48 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()
Dir_emitter.cc1703 std::vector<llvm::Value*> array_multi_index( in EmitVectorizedReduce() local
1712 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1733 array_multi_index[innermost_dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1739 llvm_ir::IrArray::Index array_index(array_multi_index, reduce->shape(), in EmitVectorizedReduce()
1766 array_multi_index[innermost_dimension] = in EmitVectorizedReduce()
1773 llvm_ir::IrArray::Index array_index(array_multi_index, reduce->shape(), in EmitVectorizedReduce()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dloop_emitter.cc101 std::vector<llvm::Value*> array_multi_index(shape_.dimensions_size()); in EmitStaticIndex() local
108 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitStaticIndex()
110 return IrArray::Index(array_multi_index, shape_, index_type); in EmitStaticIndex()
120 std::vector<llvm::Value*> array_multi_index(shape_.dimensions_size()); in EmitDynamicIndex() local
127 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitDynamicIndex()
129 return IrArray::Index(array_multi_index, shape_, index_type); in EmitDynamicIndex()