Searched refs:InputTypeForNode (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.h | 244 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
|
D | node_def_util.cc | 350 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode() function
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | memory_optimizer.cc | 704 InputTypeForNode(*node, *op_def, input_to_swap, &input_type)); in BuildSwapPair() 927 if (!InputTypeForNode(node, *op_def, input.port_id, &dtype).ok()) { in IsSwappable()
|