Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc393 input_multi_index[i] = NSWSub( in MakeElementGenerator()
Dir_emitter_unnested.cc872 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_builder_mixin.h183 llvm::Value* NSWSub(Args&&... args) { in NSWSub() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc673 input_multi_index[i] = NSWSub( in EmitElementalReduceWindow()
828 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
1008 return NSWSub(NSWAdd(strided_index, dilated_kernel_index), in EmitElementalConvolution()
1069 ? NSWSub(b_.getInt64(window.dimensions(i).size() - 1), in EmitElementalConvolution()