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.cc717 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter() local
780 Store(operand_data, selected_value_address); in HandleSelectAndScatter()
792 {Load(selected_value_address), operand_element}, "select_function"); in HandleSelectAndScatter()
803 Store(Load(operand_address), selected_value_address); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2269 llvm::Value* selected_value_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitSelectAndScatterFromMlir() local
2348 Store(operand_data, selected_value_address); in EmitSelectAndScatterFromMlir()
2369 *select_computation, {selected_value_address, operand_address}, in EmitSelectAndScatterFromMlir()
2384 Store(Load(operand_address), selected_value_address); in EmitSelectAndScatterFromMlir()