Home
last modified time | relevance | path

Searched refs:GetGeneratorForOperandIrArrays (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h187 GeneratorForOperandIrArrays GetGeneratorForOperandIrArrays( in GetGeneratorForOperandIrArrays() function
Dir_emitter_unnested.cc593 GetGeneratorForOperandIrArrays(fusion), in HandleFusion()
614 GetGeneratorForOperandIrArrays(fusion), in HandleFusion()
671 fusion, GetGeneratorForOperandIrArrays(fusion), output_array, in HandleFusion()
2003 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in BuildInitializerThunk()
2494 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(hlo), in EmitTileElementForFusion()
2677 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(unnested_hlo), in EmitPrologueForOneReduction()
2885 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(unnested_hlo), in EmitTileElementForReduction()
Dir_emitter.cc728 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h235 GeneratorForOperandIrArrays GetGeneratorForOperandIrArrays( in GetGeneratorForOperandIrArrays() function
Dir_emitter.cc2157 fusion, GetGeneratorForOperandIrArrays(fusion), GetIrArrayFor(fusion), in HandleFusion()
2163 FusedIrEmitter fused_emitter(GetGeneratorForOperandIrArrays(fusion), in HandleFusion()