Home
last modified time | relevance | path

Searched refs:OutputTypesForNode (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc440 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode() function
451 return OutputTypesForNode(node_def, op_def, outputs); in InOutTypesForNode()
457 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, op_def, &outputs)); in NumOutputsForNode()
Dnode_def_util_test.cc394 TEST(OutputTypesForNode, Simple) { in TEST() argument
403 EXPECT_TRUE(OutputTypesForNode(node_def, op_def, &types).ok()); in TEST()
Dnode_def_util.h269 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc237 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, output_dtypes)); in GetOutputDTypes()