Searched refs:InputTypeForNode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 387 EXPECT_TRUE(InputTypeForNode(node_def, op_def, 0, &type).ok()); in TEST()
389 EXPECT_TRUE(InputTypeForNode(node_def, op_def, 1, &type).ok()); in TEST()
391 EXPECT_FALSE(InputTypeForNode(node_def, op_def, 2, &type).ok()); in TEST()
|
| D | node_def_util.h | 257 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
|
| D | node_def_util.cc | 402 Status InputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypeForNode() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 727 InputTypeForNode(*node, *op_def, input_to_swap, &input_type)); in BuildSwapPair()
933 if (!InputTypeForNode(node, *op_def, input.port_id, &dtype).ok()) { in IsSwappable()
|
| D | arithmetic_optimizer.cc | 1945 TF_RETURN_IF_ERROR(InputTypeForNode(*cast_like_node, *cast_like_op_def, 0, in TrySimplify()
|