Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_builder_mixin.h191 llvm::Value* NSWSub(Args&&... args) { in NSWSub() function
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1820 NSWSub(operand_multi_index[concat_dim], source_index_phis[operand_id]); in EmitElementalConcatenate()
2649 input_multi_index[i] = NSWSub( in EmitElementalReduceWindow()
2879 return NSWSub(NSWAdd(strided_index, dilated_kernel_index), in EmitConvolution()
2940 ? NSWSub(b_->getInt64(window.dimensions(i).size() - 1), in EmitConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc747 NSWSub(NSWAdd(strided_index, window_index[i]), in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2318 operand_multi_index[i] = NSWSub(NSWAdd(strided_index, window_index[i]), in EmitSelectAndScatterFromMlir()