Home
last modified time | relevance | path

Searched defs:index_ty (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc159 IrArray::Index KernelMappingScheme::EmitBlockIndex(llvm::Type* index_ty) { in EmitBlockIndex()
218 KernelMappingScheme::EmitThreadYXCoordinate(llvm::Type* index_ty) { in EmitThreadYXCoordinate()
Dllvm_loop.h267 void SetIndexType(llvm::Type* index_ty) { in SetIndexType()
Dir_array.h59 explicit Index(llvm::Type* index_ty) : index_type_(index_ty) { in Index()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2301 llvm::Value* x, llvm::Type* index_ty) { in GetStartOffsetAndStepForX()
2319 llvm::Value* x, llvm::Type* index_ty, in EmitFullElementalTile()
2359 llvm::Value* tile_width, llvm::Type* index_ty, in EmitPartialElementalTile()
2425 llvm::Type* index_ty = tile_width->getType(); in EmitTiledElementalCodeWithBoundsCheck() local
2949 llvm::Type* index_ty) { in EmitBlock()
3081 llvm::Type* index_ty = in EmitKernel() local
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1513 llvm::Type* index_ty = sample_idx->getType(); in CalculateSampleValues() local
1630 llvm::Type* index_ty = index.GetType(); in MakePhiloxRngElementGenerator() local