Home
last modified time | relevance | path

Searched refs:null_device_ptr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc135 se::DeviceMemory<float> null_device_ptr(nullptr); in RunCudnnBatchNormForwardInferenceImpl() local
143 /*side_input=*/null_device_ptr, params->common.operand_desc, // in RunCudnnBatchNormForwardInferenceImpl()
163 se::DeviceMemory<float> null_device_ptr(nullptr); in RunCudnnBatchNormForwardTrainingImpl() local
169 /*estimated_mean=*/null_device_ptr, // in RunCudnnBatchNormForwardTrainingImpl()
170 /*estimated_variance=*/null_device_ptr, // in RunCudnnBatchNormForwardTrainingImpl()
171 /*side_input=*/null_device_ptr, // in RunCudnnBatchNormForwardTrainingImpl()
180 /*batch_mean=*/&null_device_ptr, // in RunCudnnBatchNormForwardTrainingImpl()
181 /*batch_var=*/&null_device_ptr, // in RunCudnnBatchNormForwardTrainingImpl()
192 se::DeviceMemory<float> null_device_ptr(nullptr); in RunCudnnBatchNormBackwardImpl() local
Dcudnn_batchnorm_thunk.cc101 se::DeviceMemory<float> null_device_ptr(nullptr); in ExecuteOnStream() local