Searched refs:OutputTypeForNode (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util_test.cc | 408 EXPECT_TRUE(OutputTypeForNode(node_def, op_def, 0, &type).ok()); in TEST() 410 EXPECT_TRUE(OutputTypeForNode(node_def, op_def, 1, &type).ok()); in TEST() 412 EXPECT_FALSE(OutputTypeForNode(node_def, op_def, 2, &type).ok()); in TEST()
|
D | node_def_util.h | 265 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def,
|
D | node_def_util.cc | 425 Status OutputTypeForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypeForNode() function
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | model_pruner.cc | 87 s = OutputTypeForNode(node, op_reg_data->op_def, port_id, &output_type); in IsOutputPortRefValue()
|
D | memory_optimizer.cc | 868 if (!OutputTypeForNode(node, *op_def, output.port_id, &dtype).ok()) { in IsSwappable()
|
D | arithmetic_optimizer.cc | 1948 TF_RETURN_IF_ERROR(OutputTypeForNode(*cast_like_node, *cast_like_op_def, 0, in TrySimplify()
|