Searched refs:HasFanouts (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.cc | 72 bool HasFanouts(const GraphView& graph_view, const NodeDef* node, int port) { in HasFanouts() function
|
| D | graph_view.h | 414 bool HasFanouts(const GraphView& graph_view, const NodeDef* node, int port = 0);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 361 HasFanouts(ctx.graph_view, batch_norm, 1) || // batch_mean in FindConv2DWithBatchNorm()
362 HasFanouts(ctx.graph_view, batch_norm, 2) || // batch_variance in FindConv2DWithBatchNorm()
363 HasFanouts(ctx.graph_view, batch_norm, 3) || // reserve_space_1 in FindConv2DWithBatchNorm()
364 HasFanouts(ctx.graph_view, batch_norm, 4)) // reserve_space_2 in FindConv2DWithBatchNorm()
|