Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc671 llvm::Value* strided_index = in EmitElementalReduceWindow() local
674 NSWAdd(strided_index, in EmitElementalReduceWindow()
825 llvm::Value* strided_index = in HandleSelectAndScatter() local
828 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
1004 llvm::Value* strided_index = in EmitElementalConvolution() local
1008 return NSWSub(NSWAdd(strided_index, dilated_kernel_index), in EmitElementalConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc869 llvm::Value* strided_index = NSWMul( in HandleSelectAndScatter() local
872 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()