Searched defs:BatchNorm (Results 1 – 7 of 7) sorted by relevance
27 template struct functor::BatchNorm<GPUDevice, float>; member in tensorflow::functor28 template struct functor::BatchNorm<GPUDevice, Eigen::half>; member in tensorflow::functor
27 struct BatchNorm { struct28 void operator()(const Device& d, typename TTypes<T, 4>::ConstTensor input, in operator()
23 class BatchNorm(NodeVisitor): class
25 class BatchNorm(NodeVisitor): class
91 TEST(SchemaInfoIsMutableTest, BatchNorm) { in TEST() argument
58 class BatchNorm(torch.nn.Module): class
3123 class BatchNorm(torch.nn.BatchNorm2d): class