Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc131 inv_stddev_(inv_stddev), in CudnnBatchNormBackwardThunk()
158 se::DeviceMemory<float>(buffer_allocations.GetDeviceAddress(inv_stddev_)), in ExecuteOnStream()
Dcudnn_batchnorm_thunk.h127 BufferAllocation::Slice inv_stddev_; variable