Searched refs:index_ty (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| 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 | ir_array.h | 57 explicit Index(llvm::Type* index_ty) : index_type_(index_ty) { in Index() argument
58 CHECK(index_ty->isIntegerTy()); in Index()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 520 const KernelMappingScheme& mapping_scheme, llvm::Type* index_ty,
610 llvm::Type* index_ty, mlir::Operation* unnested_hlo,
703 llvm::Value* EmitThreadId(int64 threads_per_block, llvm::Type* index_ty);
709 ThreadIdInfo EmitThreadIdInfo(int64 threads_per_block, llvm::Type* index_ty,
|
| D | ir_emitter_unnested.cc | 3943 llvm::Type* index_ty, in GetStartOffsetX() argument
3946 return llvm::ConstantInt::get(index_ty, val); in GetStartOffsetX()
3975 llvm::Type* index_ty = tile_width->getType(); in UnrollInnerTileLoop() local
3977 return llvm::ConstantInt::get(index_ty, val); in UnrollInnerTileLoop()
4006 llvm::Type* index_ty = tile_width->getType(); in EmitTile() local
4008 return llvm::ConstantInt::get(index_ty, val); in EmitTile()
4016 mapping_scheme, thread_id_info.thread_id_x, index_ty, &b_); in EmitTile()
4378 llvm::Type* index_ty, mlir::Operation* unnested_hlo, in EmitEpilogueForReduction() argument
4387 return llvm::ConstantInt::get(index_ty, c); in EmitEpilogueForReduction()
4391 EmitThreadIdInfo(mapping_scheme.GetThreadsPerBlock(), index_ty, in EmitEpilogueForReduction()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1973 RankedTensorType index_ty = index.getType().dyn_cast<RankedTensorType>(); in Verify() local
1975 if (!index_ty || !data_ty) continue; in Verify()
1977 int64_t index_rank = index_ty.getRank(); in Verify()
1979 ArrayRef<int64_t> index_shape = index_ty.getShape(); in Verify()
1985 << index_ty; in Verify()
|