Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util_test.cc373 TEST(InputTypesForNode, Simple) { in TEST() argument
382 EXPECT_TRUE(InputTypesForNode(node_def, op_def, &types).ok()); in TEST()
Dnode_def_util.cc417 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode() function
450 TF_RETURN_IF_ERROR(InputTypesForNode(node_def, op_def, inputs)); in InOutTypesForNode()
Dnode_def_util.h261 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc359 TF_RETURN_IF_ERROR(InputTypesForNode(*node, *op_def, &input_types)); in AddBranch()