Searched refs:GetConstantWithIndexType (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | sort_util.cc | 162 auto value_one = tiled_keys_index.GetConstantWithIndexType(1); in EmitTiledCompareLoop()
170 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
180 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
199 {tiled_keys_index.GetConstantWithIndexType(0), in EmitTiledCompareLoop()
210 {tiled_keys_index.GetConstantWithIndexType(0), index}); in EmitTiledCompareLoop()
225 {tiled_keys_index.GetConstantWithIndexType(0), index})); in EmitTiledCompareLoop()
241 tiled_keys_index.GetConstantWithIndexType(2)), in EmitTiledCompareLoop()
242 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
276 {tiled_keys_index.GetConstantWithIndexType(0), cache_index})); in EmitTiledCompareLoop()
|
| D | ir_array.cc | 59 auto* quot = b->CreateUDiv(linear, GetConstantWithIndexType(divisor)); in Delinearize()
62 quot, GetConstantWithIndexType(size_of_current_dimension)); in Delinearize()
159 GetConstantWithIndexType(input_shape.dimensions(i)); in SourceIndexOfReshape()
161 source_multidim_index[i] = GetConstantWithIndexType(0); in SourceIndexOfReshape()
232 llvm::Value* linear_index = GetConstantWithIndexType(0); in SourceIndexOfBitcast()
237 GetConstantWithIndexType(scale), "", in SourceIndexOfBitcast()
300 linear = builder->CreateUDiv(linear, GetConstantWithIndexType(divisor)); in SourceIndexOfBroadcast()
308 linear = builder->CreateURem(linear, GetConstantWithIndexType(mod)); in SourceIndexOfBroadcast()
318 llvm::Value* logical_linear_index = GetConstantWithIndexType(0); in Linearize()
322 builder->CreateMul((*this)[i], GetConstantWithIndexType(multiplier), "", in Linearize()
|
| D | llvm_loop.cc | 181 return AddLoop(suffix, start_index, end_index, GetConstantWithIndexType(1), in AddLoop()
216 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
217 GetConstantWithIndexType(end_index), unroll_mode, in AddLoop()
227 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
228 GetConstantWithIndexType(end_index), in AddLoop()
229 GetConstantWithIndexType(stride), unroll_mode, in AddLoop()
|
| D | kernel_tiling.cc | 63 llvm::Value* linear_index = index.GetConstantWithIndexType(0); in GetReshapedIndex()
68 index[dim], index.GetConstantWithIndexType(multiplier), "linearizing", in GetReshapedIndex()
|
| D | llvm_loop.h | 271 llvm::Constant* GetConstantWithIndexType(int64 c) const { in GetConstantWithIndexType() function
|
| D | ir_array.h | 164 llvm::Constant* GetConstantWithIndexType(int64 c) const { in GetConstantWithIndexType() function
|
| D | fused_ir_emitter.cc | 157 b_->CreateGEP(param_tile_buffer, {index.GetConstantWithIndexType(0), in HandleParameter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1792 auto concat_dim_size = source_index.GetConstantWithIndexType( in EmitElementalConcatenate()
1886 operand_multi_index.push_back(index.GetConstantWithIndexType(0)); in EmitElementalGather()
1931 index.GetConstantWithIndexType(largest_valid_start_index), in EmitElementalGather()
1932 EmitIntegralMax(index.GetConstantWithIndexType(0), in EmitElementalGather()
1952 index.GetConstantWithIndexType(i); in EmitElementalGather()
2058 return padded_index.GetConstantWithIndexType(n); in EmitElementalPad()
2290 source_multi_index[dim] = Sub(target_index.GetConstantWithIndexType( in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 374 return index.GetConstantWithIndexType(c); in MakeElementGenerator()
|
| D | ir_emitter_unnested.cc | 1111 input_window_multidim.push_back(index.GetConstantWithIndexType(0)); in EmitScatter()
1143 index.GetConstantWithIndexType(i); in EmitScatter()
1167 index.GetConstantWithIndexType(max_index))); in EmitScatter()
|