Searched refs:index_ty (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.h | 59 explicit Index(llvm::Type* index_ty) : index_type_(index_ty) { in Index() argument
60 CHECK(index_ty->isIntegerTy()); in Index()
66 llvm::Type* index_ty = nullptr)
69 index_type_ = index_ty;
75 if (index_ty != nullptr) {
76 CHECK_EQ(index_type_, index_ty);
|
| D | llvm_loop.h | 181 llvm::Type* index_ty = nullptr)
187 SetIndexType(index_ty); in name_()
267 void SetIndexType(llvm::Type* index_ty) { in SetIndexType() argument
268 index_type_ = index_ty == nullptr ? b_->getInt64Ty() : index_ty; in SetIndexType()
|
| D | kernel_tiling.cc | 159 IrArray::Index KernelMappingScheme::EmitBlockIndex(llvm::Type* index_ty) { in EmitBlockIndex() argument
165 b_->CreateIntCast(block_id, index_ty, /*isSigned=*/true, "block.id.x"); in EmitBlockIndex()
218 KernelMappingScheme::EmitThreadYXCoordinate(llvm::Type* index_ty) { in EmitThreadYXCoordinate() argument
225 b_->CreateIntCast(thread_id_raw, index_ty, in EmitThreadYXCoordinate()
228 llvm::ConstantInt::get(index_ty, GetNumberOfThreadsForDimensionX()); in EmitThreadYXCoordinate()
|
| D | kernel_tiling.h | 163 IrArray::Index EmitBlockIndex(llvm::Type* index_ty);
172 llvm::Type* index_ty);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2301 llvm::Value* x, llvm::Type* index_ty) { in GetStartOffsetAndStepForX() argument
2309 x, llvm::ConstantInt::get(index_ty, tile_size_x / num_threads_x)); in GetStartOffsetAndStepForX()
2319 llvm::Value* x, llvm::Type* index_ty, in EmitFullElementalTile() argument
2329 tile_size_x, num_threads_x, mapping_scheme, builder, x, index_ty); in EmitFullElementalTile()
2333 ksl->For(loop_name + "_y", /*start=*/llvm::ConstantInt::get(index_ty, 0), in EmitFullElementalTile()
2334 /*end=*/llvm::ConstantInt::get(index_ty, tile_size_y), in EmitFullElementalTile()
2335 /*step=*/llvm::ConstantInt::get(index_ty, num_threads_y), in EmitFullElementalTile()
2343 llvm::ConstantInt::get(index_ty, j * step_x), in EmitFullElementalTile()
2346 llvm::ConstantInt::get(index_ty, j * step_x), in EmitFullElementalTile()
2359 llvm::Value* tile_width, llvm::Type* index_ty, in EmitPartialElementalTile() argument
[all …]
|
| D | ir_emitter_unnested.h | 253 llvm::Type* index_ty);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1513 llvm::Type* index_ty = sample_idx->getType(); in CalculateSampleValues() local
1518 unsigned index_ty_size_in_bits = index_ty->getPrimitiveSizeInBits(); in CalculateSampleValues()
1630 llvm::Type* index_ty = index.GetType(); in MakePhiloxRngElementGenerator() local
1640 llvm::ConstantInt::get(index_ty, elems_per_sample); in MakePhiloxRngElementGenerator()
|