Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc724 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in HandleSelectAndScatter() local
726 Store(b_.getInt1(false), initialized_flag_address); in HandleSelectAndScatter()
762 Load(initialized_flag_address), "initialized", &b_); in HandleSelectAndScatter()
782 Store(b_.getInt1(true), initialized_flag_address); in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2279 llvm::Value* initialized_flag_address = llvm_ir::EmitAllocaAtFunctionEntry( in EmitSelectAndScatterFromMlir() local
2281 Store(b_.getInt1(false), initialized_flag_address); in EmitSelectAndScatterFromMlir()
2332 Load(initialized_flag_address), "initialized", &b_); in EmitSelectAndScatterFromMlir()
2350 Store(b_.getInt1(true), initialized_flag_address); in EmitSelectAndScatterFromMlir()