Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.h99 llvm::Value* EmitThreadId() override;
Delemental_ir_emitter.cc306 llvm::Value* GpuElementalIrEmitter::EmitThreadId() { in EmitThreadId() function in xla::gpu::GpuElementalIrEmitter
Dir_emitter_unnested.h703 llvm::Value* EmitThreadId(int64 threads_per_block, llvm::Type* index_ty);
Dir_emitter_unnested.cc4573 llvm::Value* thread_id = EmitThreadId(1024, b_.getInt32Ty()); in EmitPrintfWithThreadId()
4669 llvm::Value* IrEmitterUnnested::EmitThreadId(int64 threads_per_block, in EmitThreadId() function in xla::gpu::IrEmitterUnnested
4685 llvm::Value* thread_id = EmitThreadId(threads_per_block, index_ty); in EmitThreadIdInfo()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.h201 virtual llvm::Value* EmitThreadId() { return b_->getIntN(128, 0); } in EmitThreadId() function