Searched defs:BatchNormGrad (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | batch_norm_op_gpu.cu.cc | 29 template struct functor::BatchNormGrad<GPUDevice, float>; member in tensorflow::functor
30 template struct functor::BatchNormGrad<GPUDevice, Eigen::half>; member in tensorflow::functor
|
| H A D | batch_norm_op.h | 68 struct BatchNormGrad { struct
69 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input, in operator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | bfloat16_test.cc | 115 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormGrad)) { in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | batchnorm_expander_test.cc | 90 TEST_F(BatchNormExpanderTest, BatchNormGrad) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| H A D | operation_semantics.md | 191 ## BatchNormGrad section in Operation Semantics
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 2842 XlaOp XlaBuilder::BatchNormGrad(XlaOp operand, XlaOp scale, XlaOp batch_mean, in BatchNormGrad() function in xla::XlaBuilder
4988 XlaOp BatchNormGrad(const XlaOp operand, const XlaOp scale, in BatchNormGrad() function
|