Home
last modified time | relevance | path

Searched defs:bn_node (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpre_grad.py376 bn_node, argument
397 def add_bn_node(self, bn_node): argument
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DNodeFusionMutator.cpp87 …auto *bn_node = arm_compute::utils::cast::polymorphic_downcast<BatchNormalizationLayerNode *>(ou… in fuse_convolution_with_batch_normalization() local
167 …auto *bn_node = arm_compute::utils::cast::polymorphic_downcast<BatchNormalizationLayerNode… in fuse_depthwise_convolution_with_batch_normalization() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler_test.cc2709 const NodeState* bn_node = nullptr; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer_test.cc795 auto* bn_node = context.graph_view->GetNode("bn"); in TEST_F() local