Searched defs:offset_backprop (Results 1 – 5 of 5) sorted by relevance
147 xla::XlaOp offset_backprop; in Compile() local
172 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>()); in operator ()() local653 Tensor* offset_backprop = nullptr; in Compute() local
3286 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward()3301 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward()3317 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
3091 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward()3105 DeviceMemory<float>* offset_backprop) { in DoBatchNormalizationBackward()3120 DeviceMemory<U>* offset_backprop) { in DoBatchNormalizationBackwardImpl()
373 DeviceMemory<float> *offset_backprop) { in ThenBatchNormalizationBackward()424 DeviceMemory<float> *offset_backprop) { in ThenBatchNormalizationBackward()