Home
last modified time | relevance | path

Searched refs:ParseActivationMode (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.h35 Status ParseActivationMode(OpKernelConstruction* context,
Dfused_batch_norm_op.cc66 Status ParseActivationMode(OpKernelConstruction* context, in ParseActivationMode() function
1204 OP_REQUIRES_OK(context, ParseActivationMode(context, &activation_mode_)); in FusedBatchNormOpBase()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc670 OP_REQUIRES_OK(context, ParseActivationMode(context, &activation_mode_)); in MklFusedBatchNormOp()