Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc798 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter() local
861 Store(operand_data, selected_value_address); in HandleSelectAndScatter()
873 {Load(selected_value_address), operand_element}, "select_function"); in HandleSelectAndScatter()
884 Store(Load(operand_address), selected_value_address); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc838 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter() local
904 Store(operand_data, selected_value_address); in HandleSelectAndScatter()
920 {selected_value_address, operand_address}, select_return_buffer)); in HandleSelectAndScatter()
934 Store(Load(operand_address), selected_value_address); in HandleSelectAndScatter()