Home
last modified time | relevance | path

Searched defs:fused_batch_norm (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc97 int fused_batch_norm = kMissingIndex; member
104 int fused_batch_norm = kMissingIndex; member
154 int fused_batch_norm = kMissingIndex; member
169 int fused_batch_norm = kMissingIndex; member
842 [&](const utils::MutableNodeView& fused_batch_norm) -> bool { in IsCpuCompatibleDataType()
1007 void CopyFusedBatchNormAttributes(const NodeDef& fused_batch_norm, in IsCpuCompatibleDataType()
1210 const NodeDef& fused_batch_norm = graph->node(matched.fused_batch_norm); in IsCpuCompatibleDataType() local
1251 const NodeDef& fused_batch_norm = graph->node(matched.fused_batch_norm); in IsCpuCompatibleDataType() local
1380 const NodeDef& fused_batch_norm = graph->node(matched.fused_batch_norm); in IsCpuCompatibleDataType() local
1921 FusedBatchNorm fused_batch_norm; in IsCpuCompatibleDataType() local
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op_test.cc222 Node* fused_batch_norm; in FusedBatchNormInference() local
261 Node* fused_batch_norm; in FusedBatchNormGrad() local
Dfused_batch_norm_ex_op_test.cc579 Node* fused_batch_norm; in FusedBatchNormEx() local
/external/tensorflow/tensorflow/python/ops/
Dnn_impl.py1584 def fused_batch_norm( function
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc928 string fused_batch_norm; member