Searched refs:OutputTypesForNode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util_test.cc | 570 TEST(OutputTypesForNode, Simple) { in TEST() argument
579 EXPECT_TRUE(OutputTypesForNode(node_def, op_def, &types).ok()); in TEST()
591 TEST(OutputTypesForNode, LargeOutput) { in TEST() argument
602 EXPECT_FALSE(OutputTypesForNode(node_def, op_def, &types).ok()); in TEST()
616 EXPECT_TRUE(OutputTypesForNode(attr_slice, op_def, &types).ok()); in TEST()
|
| D | node_def_util.cc | 557 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def, in OutputTypesForNode() function
565 Status OutputTypesForNode(const AttrSlice& attrs, const OpDef& op_def, in OutputTypesForNode() function
576 return OutputTypesForNode(node_def, op_def, outputs); in InOutTypesForNode()
582 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, op_def, &outputs)); in NumOutputsForNode()
|
| D | node_def_util.h | 338 Status OutputTypesForNode(const NodeDef& node_def, const OpDef& op_def,
340 Status OutputTypesForNode(const AttrSlice& attrs, const OpDef& op_def,
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 127 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, &output_dtys)); in ExpandNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils.cc | 71 OutputTypesForNode(AttrSlice(&func.attr()), fdef->signature(), &types)); in IsPlacerInspectionRequired()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 267 TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, output_dtypes)); in GetOutputDTypes()
|