Searched refs:NSWAdd (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.cc | 325 return NSWAdd(NSWMul(block_id, threads_per_block), thread_id_in_block); in EmitThreadId()
394 NSWAdd(stridden_index, in MakeElementGenerator()
|
| D | ir_emitter_unnested.cc | 872 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_builder_mixin.h | 173 llvm::Value* NSWAdd(Args&&... args) { in NSWAdd() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 674 NSWAdd(strided_index, in EmitElementalReduceWindow()
828 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
1008 return NSWSub(NSWAdd(strided_index, dilated_kernel_index), in EmitElementalConvolution()
|