Searched refs:fused_batch_norm_ex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 265 NodeDef fused_batch_norm_ex; in RunFusedBatchNormEx() local
280 .Finalize(&fused_batch_norm_ex)); in RunFusedBatchNormEx()
286 .Input({fused_batch_norm_ex.name(), 0, t_dtype}) in RunFusedBatchNormEx()
301 .Input({fused_batch_norm_ex.name(), 3, u_dtype}) in RunFusedBatchNormEx()
302 .Input({fused_batch_norm_ex.name(), 4, u_dtype}) in RunFusedBatchNormEx()
303 .Input({fused_batch_norm_ex.name(), 5, u_dtype}) in RunFusedBatchNormEx()
321 {&fused_batch_norm_ex, &activation_grad, &fused_batch_norm_grad}); in RunFusedBatchNormEx()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1008 NodeDef* fused_batch_norm_ex) { in IsCpuCompatibleDataType() argument
1012 auto* attr = fused_batch_norm_ex->mutable_attr(); in IsCpuCompatibleDataType()
1911 FusedBatchNormEx fused_batch_norm_ex; in IsCpuCompatibleDataType() local
1913 FindFusedBatchNormEx(ctx, i, &fused_batch_norm_ex)) { in IsCpuCompatibleDataType()
1915 &ctx, fused_batch_norm_ex, &invalidated_nodes, &nodes_to_delete)); in IsCpuCompatibleDataType()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 271 csinfo_.fused_batch_norm_ex = "_FusedBatchNormEx"; in MklLayoutRewritePass()
483 rinfo_.push_back({csinfo_.fused_batch_norm_ex, in MklLayoutRewritePass()
930 string fused_batch_norm_ex; member
3748 n->type_string() != csinfo_.fused_batch_norm_ex && in CheckForNodeRewrite()
|