Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.h267 void SetIndexType(llvm::Type* index_ty) { in SetIndexType()
Dir_array.h57 explicit Index(llvm::Type* index_ty) : index_type_(index_ty) { in Index()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3943 llvm::Type* index_ty, in GetStartOffsetX()
3975 llvm::Type* index_ty = tile_width->getType(); in UnrollInnerTileLoop() local
4006 llvm::Type* index_ty = tile_width->getType(); in EmitTile() local
4378 llvm::Type* index_ty, mlir::Operation* unnested_hlo, in EmitEpilogueForReduction()
4670 llvm::Type* index_ty) { in EmitThreadId()
4681 int64 threads_per_block, llvm::Type* index_ty, int64 num_threads_x) { in EmitThreadIdInfo()
4695 const KernelMappingScheme& mapping_scheme, llvm::Type* index_ty, in EmitTilingKernel()
4745 llvm::Type* index_ty = block_coords.GetType(); in EmitTilingKernel() local
5430 llvm::Type* index_ty = GetIndexTypeForKernelFromMlir( in EmitIRForReduction() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1973 RankedTensorType index_ty = index.getType().dyn_cast<RankedTensorType>(); in Verify() local