Home
last modified time | relevance | path

Searched defs:batch_var (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op_test.cc107 Tensor batch_var; in VerifyTensorsClose() local
232 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
278 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler_test.cc898 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithBatchNorm() local
2235 auto batch_var = batch_norm.batch_variance; in CreateGrapplerItemWithInterDeviceTransfers() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1550 void* batch_mean, void* batch_var, in SetBatchNormForwardArgs()
3646 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3666 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
3688 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1171 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
1191 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
H A Dstream.cc337 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
391 DeviceMemory<float> *batch_mean, DeviceMemory<float> *batch_var, in ThenBatchNormalizationForward()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc1361 Tensor* batch_var = nullptr; in ComputeWithReservedSpace() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc5417 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
5441 DeviceMemory<float>* batch_mean, DeviceMemory<float>* batch_var, in DoBatchNormalizationForward()
5466 DeviceMemory<U>* batch_mean, DeviceMemory<U>* batch_var, in DoBatchNormalizationForwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2843 XlaOp batch_var, XlaOp grad_output, in BatchNormGrad()
4989 const XlaOp batch_mean, const XlaOp batch_var, in BatchNormGrad()