Searched refs:InputTypesForNode (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 128 TF_RETURN_IF_ERROR(InputTypesForNode(node_def, *op_def, &input_dtys)); in ExpandNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 549 TEST(InputTypesForNode, Simple) { in TEST() argument
558 EXPECT_TRUE(InputTypesForNode(node_def, op_def, &types).ok()); in TEST()
|
| D | node_def_util.cc | 546 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in InputTypesForNode() function
588 TF_RETURN_IF_ERROR(InputTypesForNode(node_def, op_def, inputs)); in InOutTypesForNode()
|
| D | node_def_util.h | 346 Status InputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
|