Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h470 const llvm_ir::ElementGenerator& scatter_indices_gen,
486 llvm_ir::ElementGenerator scatter_indices_gen; member
Dir_emitter_unnested.cc1970 desc.scatter_indices_gen, in HandleFusion()
2567 const llvm_ir::ElementGenerator& scatter_indices_gen, in EmitScatter() argument
2588 desc.scatter_indices_gen = scatter_indices_gen; in EmitScatter()
2672 desc.scatter_indices_gen(raw_scatter_index_index.SourceIndexOfReshape( in EmitScatter()