Searched defs:batch_var (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 107 Tensor batch_var; in VerifyTensorsClose() local
232 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
278 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 891 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm() local
2228 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1354 Tensor* batch_var = nullptr; in ComputeWithReservedSpace() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1547 void* batch_mean, void* batch_var, in SetBatchNormForwardArgs()
3525 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3545 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3567 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1128 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
1148 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
|
| D | stream.cc | 351 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
402 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4566 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
4590 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
4615 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2717 XlaOp batch_var, XlaOp grad_output, in BatchNormGrad()
4817 const XlaOp batch_mean, const XlaOp batch_var, in BatchNormGrad()
|