Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc802 llvm::Value* selected_index_address = in HandleSelectAndScatter() local
852 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
895 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc842 llvm::Value* selected_index_address = in HandleSelectAndScatter() local
895 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
946 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()