Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc316 return NSWAdd(NSWMul(block_id, threads_per_block), thread_id_in_block); in EmitThreadId()
Dir_emitter_unnested.cc2317 NSWMul(source_index[i], index_typed_constant(stride)); in EmitSelectAndScatterFromMlir()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_builder_mixin.h186 llvm::Value* NSWMul(Args&&... args) { in NSWMul() function
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc2648 NSWMul(index[i], index_typed_const(window.dimensions(i).stride())); in EmitElementalReduceWindow()
2652 NSWMul(window_index[i], in EmitElementalReduceWindow()
2876 NSWMul(output_index, b_->getInt64(window_dim.stride())); in EmitConvolution()
2878 NSWMul(kernel_index, b_->getInt64(window_dim.window_dilation())); in EmitConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc745 NSWMul(source_index[i], b_.getInt64(window.dimensions(i).stride())); in HandleSelectAndScatter()