Searched refs:element_address (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 53 element_address, in EmitCompareLoopBody() argument
116 values_to_compare.push_back(element_address(i, compare_keys_index)); in EmitCompareLoopBody()
117 values_to_compare.push_back(element_address(i, current_keys_index)); in EmitCompareLoopBody()
207 auto element_address = [&](int64 operand, llvm::Value* index) { in EmitTiledCompareLoop() local
248 element_address, write_element, emit_compare_callback, b); in EmitTiledCompareLoop()
253 tiled_keys_index.GetType(), element_address, write_element, in EmitTiledCompareLoop()
260 tiled_keys_index.GetType(), element_address, write_element, in EmitTiledCompareLoop()
367 auto element_address = [&](int64 operand, llvm::Value* index) { in EmitSortInPlace() local
384 xor_masks[0], tiles_index.GetType(), element_address, write_element, in EmitSortInPlace()
|
| D | ir_array.cc | 395 llvm::Value* element_address = in EmitReadArrayElement() local
397 llvm::LoadInst* load = b->CreateLoad(element_address); in EmitReadArrayElement()
405 llvm::Value* element_address = in EmitWriteArrayElement() local
407 llvm::StoreInst* store = b->CreateStore(value, element_address); in EmitWriteArrayElement()
|