Home
last modified time | relevance | path

Searched refs:source_multi_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc183 std::vector<llvm::Value*> source_multi_index(multidim_.size()); in SourceIndexOfSlice() local
189 source_multi_index[i] = builder->CreateAdd( in SourceIndexOfSlice()
194 source_multi_index[i] = builder->CreateAdd( in SourceIndexOfSlice()
198 return Index(source_multi_index, operand_shape, index_type_); in SourceIndexOfSlice()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1786 std::vector<llvm::Value*> source_multi_index = source_index.multidim(); in EmitElementalConcatenate() local
1795 source_index_phis[operand_id]->addIncoming(source_multi_index[concat_dim], in EmitElementalConcatenate()
1797 CondBr(ICmpULT(source_multi_index[concat_dim], concat_dim_size), in EmitElementalConcatenate()
1803 source_multi_index[concat_dim] = in EmitElementalConcatenate()
1804 Sub(source_multi_index[concat_dim], concat_dim_size); in EmitElementalConcatenate()
2288 std::vector<llvm::Value*> source_multi_index = target_index.multidim(); in MakeElementGenerator() local
2290 source_multi_index[dim] = Sub(target_index.GetConstantWithIndexType( in MakeElementGenerator()
2295 source_multi_index, operand->shape(), target_index.GetType()); in MakeElementGenerator()