Home
last modified time | relevance | path

Searched defs:BatchNormGrad (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op_gpu.cu.cc29 template struct functor::BatchNormGrad<GPUDevice, float>; member in tensorflow::functor
30 template struct functor::BatchNormGrad<GPUDevice, Eigen::half>; member in tensorflow::functor
Dbatch_norm_op.h74 struct BatchNormGrad { struct
75 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input, in operator()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbfloat16_test.cc117 XLA_TEST_F(Bfloat16Test, DISABLED_ON_INTERPRETER(BatchNormGrad)) { in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander_test.cc91 TEST_F(BatchNormExpanderTest, BatchNormGrad) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md191 ## BatchNormGrad section in Operation Semantics
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc254 TEST_F(LayoutAssignmentTest, BatchNormGrad) { in TEST_F() argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2688 XlaOp XlaBuilder::BatchNormGrad(XlaOp operand, XlaOp scale, XlaOp batch_mean, in BatchNormGrad() function in xla::XlaBuilder
4954 XlaOp BatchNormGrad(const XlaOp operand, const XlaOp scale, in BatchNormGrad() function