Searched refs:has_side_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 141 bool has_side_input, const string& activation_mode, in RunFusedBatchNorm() argument
169 if (has_side_input) { in RunFusedBatchNorm()
230 bool has_side_input, const string& activation_mode, in RunFusedBatchNormEx() argument
260 if (has_side_input) { in RunFusedBatchNormEx()
429 bool has_side_input, in VerifyFusedBatchNormEx() argument
440 has_side_input, activation_mode, fwd, bwd); in VerifyFusedBatchNormEx()
452 has_side_input, activation_mode, fwd, bwd); in VerifyFusedBatchNormEx()
571 bool has_side_input = side_input_and_activation.first; in FusedBatchNormEx() local
574 if (has_side_input) { in FusedBatchNormEx()
|
| D | fused_batch_norm_op.cc | 829 const bool has_side_input = side_input != nullptr; in operator ()() local
833 if (!is_training && (has_side_input || has_activation)) { in operator ()()
837 if (has_side_input) { in operator ()()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | mkl_remapper_test.cc | 317 for (bool has_side_input : {true, false}) { in TEST_F()
342 if (has_side_input) { in TEST_F()
366 if (has_side_input) in TEST_F()
388 if (has_side_input) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 244 int oy, int oz, bool has_side_input, in DescribeFusedConv2DBiasActivation() argument
281 if (has_side_input) { in DescribeFusedConv2DBiasActivation()
|