Searched refs:grad_data_size (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/resampler/kernels/
H A Dresampler_ops_gpu.cu.cc252 const int grad_data_size = local
261 config = ::tensorflow::GetCudaLaunchConfig(grad_data_size, d);
264 grad_data_size, grad_data);
H A Dresampler_ops.cc217 const int grad_data_size = local
219 memset(grad_data, 0, sizeof(T) * grad_data_size);

Completed in 124 milliseconds