Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc851 llvm::Value* selected_index_address_slot = in HandleSelectAndScatter() local
853 Store(operand_index[i], selected_index_address_slot); in HandleSelectAndScatter()
894 llvm::Value* selected_index_address_slot = in HandleSelectAndScatter() local
896 selected_multi_index.push_back(Load(selected_index_address_slot)); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc894 llvm::Value* selected_index_address_slot = in HandleSelectAndScatter() local
896 Store(operand_index[i], selected_index_address_slot); in HandleSelectAndScatter()
945 llvm::Value* selected_index_address_slot = in HandleSelectAndScatter() local
947 selected_multi_index.push_back(Load(selected_index_address_slot)); in HandleSelectAndScatter()