Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc64 auto* quot = b->CreateUDiv(linear, GetConstantWithIndexType(divisor)); in Delinearize()
67 quot, GetConstantWithIndexType(size_of_current_dimension)); in Delinearize()
81 llvm::Value* divisor = GetConstantWithIndexType(1); in Delinearize()
217 source_multidim_index[i] = GetConstantWithIndexType(0); in SourceIndexOfReshape()
244 GetConstantWithIndexType(input_shape.dimensions(i)); in SourceIndexOfReshape()
246 source_multidim_index[i] = GetConstantWithIndexType(0); in SourceIndexOfReshape()
275 builder->CreateMul(multidim_[i], GetConstantWithIndexType(stride)), in SourceIndexOfSlice()
276 GetConstantWithIndexType(starts[i])); in SourceIndexOfSlice()
279 builder->CreateAdd(multidim_[i], GetConstantWithIndexType(starts[i])); in SourceIndexOfSlice()
323 llvm::Value* linear_index = GetConstantWithIndexType(0); in SourceIndexOfBitcast()
[all …]
Dsort_util.cc163 auto value_one = tiled_keys_index.GetConstantWithIndexType(1); in EmitTiledCompareLoop()
171 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
181 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
200 {tiled_keys_index.GetConstantWithIndexType(0), in EmitTiledCompareLoop()
211 {tiled_keys_index.GetConstantWithIndexType(0), index}); in EmitTiledCompareLoop()
226 {tiled_keys_index.GetConstantWithIndexType(0), index})); in EmitTiledCompareLoop()
242 tiled_keys_index.GetConstantWithIndexType(2)), in EmitTiledCompareLoop()
243 tiled_keys_index.GetConstantWithIndexType( in EmitTiledCompareLoop()
278 {tiled_keys_index.GetConstantWithIndexType(0), cache_index})); in EmitTiledCompareLoop()
Dllvm_loop.cc182 return AddLoop(suffix, start_index, end_index, GetConstantWithIndexType(1), in AddLoop()
217 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
218 GetConstantWithIndexType(end_index), unroll_mode, in AddLoop()
228 return AddLoop(suffix, GetConstantWithIndexType(start_index), in AddLoop()
229 GetConstantWithIndexType(end_index), in AddLoop()
230 GetConstantWithIndexType(stride), unroll_mode, in AddLoop()
Dllvm_loop.h271 llvm::Constant* GetConstantWithIndexType(int64 c) const { in GetConstantWithIndexType() function
Dir_array.h166 llvm::Constant* GetConstantWithIndexType(int64 c) const { in GetConstantWithIndexType() function
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1842 source_index.GetConstantWithIndexType(concat_dim_size), in EmitElementalConcatenate()
1846 source_index.GetConstantWithIndexType(concat_dim_size)), in EmitElementalConcatenate()
1933 operand_multi_index.push_back(index.GetConstantWithIndexType(0)); in EmitElementalGather()
2011 index.GetConstantWithIndexType(i); in EmitElementalGather()
2117 return padded_index.GetConstantWithIndexType(n); in EmitElementalPad()
2334 source_multi_index[dim] = Sub(target_index.GetConstantWithIndexType( in MakeElementGenerator()
2629 return index.GetConstantWithIndexType(c); in EmitElementalReduceWindow()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2628 input_window_multidim.push_back(index.GetConstantWithIndexType(0)); in EmitScatter()
2663 index.GetConstantWithIndexType(i); in EmitScatter()
2689 index.GetConstantWithIndexType(max_index))); in EmitScatter()
4169 {index.GetConstantWithIndexType(0), x_loc, y_loc}), in EmitTileElementForFusion()
4372 llvm::Value* x_dim_size = index.GetConstantWithIndexType(dims_in_elem[kDimX]); in GetUntransposedOutputLinearAddress()
5764 index.GetConstantWithIndexType(slice->slice_starts(dim))); in EmitElementForInputFusibleSlices()
5767 index.GetConstantWithIndexType(slice->slice_limits(dim))); in EmitElementForInputFusibleSlices()
5780 index.GetConstantWithIndexType(slice->slice_starts(dim))); in EmitElementForInputFusibleSlices()