HomeSort by relevance Sort by last modified time
    Searched refs:shared_memory_per_block (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
device_description.h 152 int64 shared_memory_per_block() const { return shared_memory_per_block_; } function in class:stream_executor::DeviceDescription
  /external/tensorflow/tensorflow/stream_executor/gpu/
gpu_executor.h 79 uint64 shared_memory_per_block,
86 uint64 shared_memory_per_block,
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc 510 uint64 shared_memory_per_block,
517 shared_memory_per_block, 0);
527 uint64 shared_memory_per_block,
534 shared_memory_per_block, 0);
    [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_gpu_executor.cc 346 uint64 shared_memory_per_block,
356 uint64 shared_memory_per_block,
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc     [all...]

Completed in 78 milliseconds