Home
last modified time | relevance | path

Searched defs:threads_per_block (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.h41 LaunchDimensions(int64 block_count, int64 threads_per_block) in LaunchDimensions()
49 int64 threads_per_block() const { return threads_per_block_; } in threads_per_block() function
Dpartition_assignment.cc66 int64 threads_per_block; in CalculateLaunchDimensions() local
Delemental_ir_emitter.cc308 llvm::Value* threads_per_block = ir_builder_->CreateIntCast( in EmitThreadId() local
/external/tensorflow/tensorflow/core/kernels/
Dadjust_saturation_op_gpu.cu.cc35 const int threads_per_block = config.thread_per_block; in operator ()() local
Dadjust_hue_op_gpu.cu.cc33 const int threads_per_block = config.thread_per_block; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc143 uint64 element_count, uint64 *threads_per_block, in CalculateDimensionality()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_driver.cc1612 CudaContext* context, CUfunction kernel, int threads_per_block, in GetMaxOccupiedBlocksPerCore()