| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 227 std::function<const DeviceMemory<float>&()> var_to_inv_var,
240 std::function<const DeviceMemory<float>&()> var_to_inv_var,
568 std::function<const DeviceMemory<U>&()> var_to_inv_var,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 218 std::function<const DeviceMemory<float>&()> var_to_inv_var,
231 std::function<const DeviceMemory<float>&()> var_to_inv_var,
[all...] |
| rocm_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 338 std::function<const DeviceMemory<U>&()> var_to_inv_var = local
365 is_training, std::move(var_to_inv_var),
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
| stream.h | 242 std::function<const DeviceMemory<float> &()> var_to_inv_var,
263 std::function<const DeviceMemory<float> &()> var_to_inv_var,
[all...] |
| stream.cc | 349 std::function<const DeviceMemory<float> &()> var_to_inv_var,
358 saved_inv_var, is_training, std::move(var_to_inv_var),
399 std::function<const DeviceMemory<float> &()> var_to_inv_var,
408 saved_inv_var, is_training, std::move(var_to_inv_var),
[all...] |
