Home
last modified time | relevance | path

Searched refs:TiledParameterInfo (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h90 void SetTiledParameterInfo(const llvm_ir::TiledParameterInfo* info) { in SetTiledParameterInfo()
121 const llvm_ir::TiledParameterInfo* tiled_parameter_info_;
Dkernel_tiling.h213 class TiledParameterInfo {
215 TiledParameterInfo(absl::Span<llvm::Value* const> param_buffers, in TiledParameterInfo() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h78 void SetTiledParamInfo(llvm_ir::TiledParameterInfo* tiled_param_info) { in SetTiledParamInfo()
86 llvm_ir::TiledParameterInfo* GetTiledParameterInfo() const { in GetTiledParameterInfo()
93 llvm_ir::TiledParameterInfo* tiled_param_info_;
Dir_emitter_unnested.cc2459 llvm_ir::TiledParameterInfo* tiled_param_info = in EmitTileElementForCopy()
2489 llvm_ir::TiledParameterInfo* tiled_param_info = in EmitTileElementForFusion()
2860 llvm_ir::TiledParameterInfo* tiled_param_info = in EmitTileElementForReduction()
3172 llvm_ir::TiledParameterInfo tiled_param_info(param_shmem_buffers, y, x); in EmitKernel()