Searched refs:FusedChildContext (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1199 FusedChildContext(op_context, "Conv2D", output, {conv_input, filter}), in PredictFusedConv2DBiasActivation()
1200 FusedChildContext(op_context, "Mul", output, {output, conv_input_scale}), in PredictFusedConv2DBiasActivation()
1201 FusedChildContext(op_context, "BiasAdd", output, {output, bias}), in PredictFusedConv2DBiasActivation()
1202 FusedChildContext(op_context, "Relu", output, {output})}; in PredictFusedConv2DBiasActivation()
1206 component_ops.push_back(FusedChildContext(op_context, "Mul", side_input, in PredictFusedConv2DBiasActivation()
1209 FusedChildContext(op_context, "Add", output, {side_input, output})); in PredictFusedConv2DBiasActivation()
1399 OpContext OpLevelCostEstimator::FusedChildContext( in FusedChildContext() function in tensorflow::grappler::OpLevelCostEstimator
|
| D | op_level_cost_estimator.h | 176 static OpContext FusedChildContext(
|