Home
last modified time | relevance | path

Searched refs:EmitTargetElementLoop (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.h57 Status EmitTargetElementLoop(
Dir_emitter_nested.cc116 Status IrEmitterNested::EmitTargetElementLoop( in EmitTargetElementLoop() function in xla::gpu::IrEmitterNested
Dir_emitter.cc75 return EmitTargetElementLoop( in DefaultAction()
673 return EmitTargetElementLoop( in HandleReduce()
732 return EmitTargetElementLoop(*fusion, fused_emitter.GetRootGenerator()); in HandleFusion()
Dir_emitter.h149 virtual Status EmitTargetElementLoop(
Dir_emitter_unnested.h184 Status EmitTargetElementLoop(
Dir_emitter_unnested.cc2242 Status IrEmitterUnnested::EmitTargetElementLoop( in EmitTargetElementLoop() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h310 Status EmitTargetElementLoop(
313 Status EmitTargetElementLoop(
Dir_emitter.cc783 TF_RETURN_IF_ERROR(EmitTargetElementLoop( in HandleSelectAndScatter()
2104 TF_RETURN_IF_ERROR(EmitTargetElementLoop( in HandlePad()
2167 return EmitTargetElementLoop(fusion, fused_emitter.GetRootGenerator()); in HandleFusion()
2661 TF_RETURN_IF_ERROR(EmitTargetElementLoop( in HandleRng()
2942 Status IrEmitter::EmitTargetElementLoop( in EmitTargetElementLoop() function in xla::cpu::IrEmitter
2945 return EmitTargetElementLoop(target_op, /*desc=*/"", element_generator); in EmitTargetElementLoop()
2948 Status IrEmitter::EmitTargetElementLoop( in EmitTargetElementLoop() function in xla::cpu::IrEmitter
3035 return EmitTargetElementLoop( in DefaultAction()