Searched refs:FusedBatchNormActivationMode (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 31 enum class FusedBatchNormActivationMode { kIdentity, kRelu }; enum
33 std::string ToString(FusedBatchNormActivationMode activation_mode);
36 FusedBatchNormActivationMode* activation_mode);
57 FusedBatchNormActivationMode activation_mode,
|
| D | fused_batch_norm_op.cu.cc | 128 bool add_side_input, FusedBatchNormActivationMode activation_mode,
164 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
166 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
177 FusedBatchNormActivationMode activation_mode>
247 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
250 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
279 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
282 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
297 bool add_side_input, FusedBatchNormActivationMode activation_mode>
326 FusedBatchNormActivationMode activation_mode, in operator ()()
[all …]
|
| D | fused_batch_norm_op.cc | 57 string ToString(FusedBatchNormActivationMode activation_mode) { in ToString()
59 case FusedBatchNormActivationMode::kIdentity: in ToString()
61 case FusedBatchNormActivationMode::kRelu: in ToString()
67 FusedBatchNormActivationMode* activation_mode) { in ParseActivationMode()
72 *activation_mode = FusedBatchNormActivationMode::kIdentity; in ParseActivationMode()
76 *activation_mode = FusedBatchNormActivationMode::kRelu; in ParseActivationMode()
98 FusedBatchNormActivationMode activation_mode, in operator ()()
108 activation_mode == FusedBatchNormActivationMode::kIdentity, in operator ()()
236 FusedBatchNormActivationMode activation_mode, in operator ()()
246 activation_mode == FusedBatchNormActivationMode::kIdentity, in operator ()()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 41 using FusedBNActivationMode = functor::FusedBatchNormActivationMode;
|