Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc819 bool GpuExecutor::FillBlockDimLimit(BlockDim* block_dim_limit) const { in FillBlockDimLimit()
829 block_dim_limit->x = x; in FillBlockDimLimit()
830 block_dim_limit->y = y; in FillBlockDimLimit()
831 block_dim_limit->z = z; in FillBlockDimLimit()
927 BlockDim block_dim_limit; in PopulateDeviceDescription() local
928 FillBlockDimLimit(&block_dim_limit); in PopulateDeviceDescription()
929 builder.set_block_dim_limit(block_dim_limit); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc917 bool GpuExecutor::FillBlockDimLimit(BlockDim* block_dim_limit) const { in FillBlockDimLimit()
927 block_dim_limit->x = x; in FillBlockDimLimit()
928 block_dim_limit->y = y; in FillBlockDimLimit()
929 block_dim_limit->z = z; in FillBlockDimLimit()
1102 BlockDim block_dim_limit; in PopulateDeviceDescription() local
1103 FillBlockDimLimit(&block_dim_limit); in PopulateDeviceDescription()
1104 builder.set_block_dim_limit(block_dim_limit); in PopulateDeviceDescription()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc73 const BlockDim &block_dim = block_dim_limit(); in ToMap()
Ddevice_description.h79 const BlockDim &block_dim_limit() const { return block_dim_limit_; } in block_dim_limit() function
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h206 bool FillBlockDimLimit(BlockDim* block_dim_limit) const;