Searched refs:FusedChildContext (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.h | 299 static OpContext FusedChildContext(
|
| D | op_level_cost_estimator.cc | 1763 FusedChildContext(op_context, "Conv2D", output, {conv_input, filter}), in PredictFusedConv2DBiasActivation()
1764 FusedChildContext(op_context, "Mul", output, {output, conv_input_scale}), in PredictFusedConv2DBiasActivation()
1765 FusedChildContext( in PredictFusedConv2DBiasActivation()
1768 FusedChildContext(op_context, "Relu", output, {output})}; in PredictFusedConv2DBiasActivation()
1772 component_ops.push_back(FusedChildContext(op_context, "Mul", side_input, in PredictFusedConv2DBiasActivation()
1774 component_ops.push_back(FusedChildContext( in PredictFusedConv2DBiasActivation()
2083 OpContext OpLevelCostEstimator::FusedChildContext( in FusedChildContext() function in tensorflow::grappler::OpLevelCostEstimator
|