| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 210 std::function<const DeviceMemory<float>&()> var_to_inv_var,
223 std::function<const DeviceMemory<float>&()> var_to_inv_var,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 317 std::function<const DeviceMemory<U>&()> var_to_inv_var = local
344 is_training, std::move(var_to_inv_var),
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 914 // var_to_inv_var: a function to convert the variance to inverted variance
[all...] |
| stream.h | 231 std::function<const DeviceMemory<float> &()> var_to_inv_var,
252 std::function<const DeviceMemory<float> &()> var_to_inv_var,
[all...] |
| stream.cc | 324 std::function<const DeviceMemory<float> &()> var_to_inv_var,
333 saved_inv_var, is_training, std::move(var_to_inv_var),
374 std::function<const DeviceMemory<float> &()> var_to_inv_var,
383 saved_inv_var, is_training, std::move(var_to_inv_var),
[all...] |
