Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc146 const IrArray::Index& tiled_keys_index, int64 dimension_to_sort, in EmitTiledCompareLoop() argument
157 thread_id = b->CreateIntCast(thread_id, tiled_keys_index.GetType(), in EmitTiledCompareLoop()
163 auto value_one = tiled_keys_index.GetConstantWithIndexType(1); in EmitTiledCompareLoop()
165 b->CreateShl(tiled_keys_index[dimension_to_sort], value_one); in EmitTiledCompareLoop()
171 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
181 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
191 std::vector<llvm::Value*> keys_multi_index = tiled_keys_index.multidim(); in EmitTiledCompareLoop()
196 tiled_keys_index.GetType()); in EmitTiledCompareLoop()
200 {tiled_keys_index.GetConstantWithIndexType(0), in EmitTiledCompareLoop()
211 {tiled_keys_index.GetConstantWithIndexType(0), index}); in EmitTiledCompareLoop()
[all …]