Searched refs:input_to_broadcast (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 104 bool ReplaceOperationWithBroadcastTo(int input_to_broadcast,
115 void ReplaceBinaryOperationWithBroadcastTo(int input_to_broadcast,
|
| D | constant_folding.cc | 1861 int input_to_broadcast, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithBroadcastTo() argument
1874 *node, strings::StrCat("-broadcastto_shape-", input_to_broadcast)); in ReplaceOperationWithBroadcastTo()
1892 if (i != input_to_broadcast) { in ReplaceOperationWithBroadcastTo()
1907 node->mutable_input()->SwapElements(0, input_to_broadcast); in ReplaceOperationWithBroadcastTo()
2008 int input_to_broadcast, const GraphProperties& properties, NodeDef* node, in ReplaceBinaryOperationWithBroadcastTo() argument
2010 if (!ReplaceOperationWithBroadcastTo(input_to_broadcast, properties, node, in ReplaceBinaryOperationWithBroadcastTo()
|