Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc325 return NSWAdd(NSWMul(block_id, threads_per_block), thread_id_in_block); in EmitThreadId()
391 llvm::Value* stridden_index = NSWMul( in MakeElementGenerator()
395 NSWMul(window_index[i], in MakeElementGenerator()
Dir_emitter_unnested.cc869 llvm::Value* strided_index = NSWMul( in HandleSelectAndScatter()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_builder_mixin.h178 llvm::Value* NSWMul(Args&&... args) { in NSWMul() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc672 NSWMul(index[i], b_.getInt64(window.dimensions(i).stride())); in EmitElementalReduceWindow()
675 NSWMul(window_index[i], in EmitElementalReduceWindow()
826 NSWMul(source_index[i], b_.getInt64(window.dimensions(i).stride())); in HandleSelectAndScatter()
1005 NSWMul(output_index, b_.getInt64(window_dim.stride())); in EmitElementalConvolution()
1007 NSWMul(kernel_index, b_.getInt64(window_dim.window_dilation())); in EmitElementalConvolution()