Searched defs:batch_var (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 86 typename TTypes<U>::Vec batch_var(batch_var_output->vec<U>()); in operator ()() local
577 Tensor* batch_var = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 274 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm() local
1562 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1383 void* batch_mean, void* batch_var, in SetBatchNormForwardArgs()
3004 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3023 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3043 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3186 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3207 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3229 DeviceMemory<T>* y, DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2095 const XlaOp& batch_mean, const XlaOp& batch_var, in BatchNormGrad()
3502 const XlaOp& batch_mean, const XlaOp& batch_var, in BatchNormGrad()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 347 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward()
397 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward()
|