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.cc42 std::vector<llvm::Value*> array_multi_index(num_dims); in EmitIndexAndSetExitBasicBlock() local
57 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
64 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
74 llvm_ir::IrArray::Index array_index(array_multi_index, shape_, index_type); in EmitIndexAndSetExitBasicBlock()
Dir_emitter.cc1723 std::vector<llvm::Value*> array_multi_index( in EmitVectorizedReduce() local
1732 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1753 array_multi_index[innermost_dimension] = loop->GetIndVarValue(); in EmitVectorizedReduce()
1759 llvm_ir::IrArray::Index array_index(array_multi_index, reduce->shape(), in EmitVectorizedReduce()
1786 array_multi_index[innermost_dimension] = in EmitVectorizedReduce()
1793 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 EmitIndexAndSetExitBasicBlock() local
108 array_multi_index[dimension] = loop->GetIndVarValue(); in EmitIndexAndSetExitBasicBlock()
110 IrArray::Index array_index(array_multi_index, shape_, index_type); in EmitIndexAndSetExitBasicBlock()