Searched refs:InferOutputType (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 253 StatusOr<mlir::Type> InferOutputType(const Node& node, int idx,
955 StatusOr<mlir::Type> ImporterBase::InferOutputType(const Node& node, int idx, in InferOutputType() function in tensorflow::__anon83888ea00111::ImporterBase
1890 TF_ASSIGN_OR_RETURN(auto type, InferOutputType(node, i, builder_)); in ConvertNode()
2110 InferOutputType(*node, /*idx=*/0, builder)); in InferLibFunctionType()
2416 InferOutputType(*ret.node, ret.index, builder)); in InferMainFunctionType()
2466 InferOutputType(*arg_node.node, /*idx=*/0, builder)); in GetArgsRetsAndTypesFromFunctionGraph()
|