Searched refs:input_to_broadcast (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 107 bool ReplaceOperationWithBroadcastTo(int input_to_broadcast,
118 void ReplaceBinaryOperationWithBroadcastTo(int input_to_broadcast,
|
| D | constant_folding.cc | 1812 int input_to_broadcast, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithBroadcastTo() argument
1825 *node, strings::StrCat("-broadcastto_shape-", input_to_broadcast)); in ReplaceOperationWithBroadcastTo()
1843 if (i != input_to_broadcast) { in ReplaceOperationWithBroadcastTo()
1858 node->mutable_input()->SwapElements(0, input_to_broadcast); in ReplaceOperationWithBroadcastTo()
1959 int input_to_broadcast, const GraphProperties& properties, NodeDef* node, in ReplaceBinaryOperationWithBroadcastTo() argument
1961 if (!ReplaceOperationWithBroadcastTo(input_to_broadcast, properties, node, in ReplaceBinaryOperationWithBroadcastTo()
|