Searched refs:InitializeBuffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.h | 87 void InitializeBuffer(se::Stream* stream, PrimitiveType buffer_type,
|
| D | gemm_algorithm_picker.cc | 83 InitializeBuffer(stream, op->shape().element_type(), &rng_state, buffer); in DoUncachedGemmAutotune()
125 InitializeBuffer(stream, gemm->shape().element_type(), &rng_state, in DoUncachedGemmAutotune()
|
| D | stream_executor_util.cc | 304 void InitializeBuffer(se::Stream* stream, PrimitiveType buffer_type, in InitializeBuffer() function
|
| D | gpu_conv_algorithm_picker.cc | 398 InitializeBuffer(stream, buffer_shape.element_type(), &rng_state, buffer); in PickBestAlgorithmNoCacheCuda()
|