in tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/internal/c_api/presets/tensorflow.java [289:519]
public void map(InfoMap infoMap) {
infoMap
.put(
new Info("c_api_internal.h")
.linePatterns(
"struct TF_OperationDescription \\{",
"\\};",
"struct TF_Graph \\{",
"\\};",
"struct TF_Operation \\{",
"\\};",
"// Exposed helper functions",
"// End Exposed helper functions"))
.put(
new Info("graph.h")
.linePatterns(
"class Node \\{", "// Stores debug information associated with the Node."))
.put(new Info("Node").cppTypes("tensorflow::Node").purify())
.put(
new Info(
"tensorflow::NodeDef",
"tensorflow::OpDef",
"tensorflow::AttrSlice",
"tensorflow::Edge",
"tensorflow::EdgeSet",
"tensorflow::WhileContext",
"tensorflow::NodeProperties",
"protobuf::RepeatedPtrField",
"gtl::iterator_range<NeighborIter>",
"tensorflow::DataType",
"tensorflow::DataTypeVector",
"tensorflow::Node::set_original_func_names",
"tensorflow::Node::set_original_node_names",
"tensorflow::Node::AddAttr",
"tensorflow::Node::ClearAttr",
"tensorflow::Node::input_node")
.skip())
.put(
new Info("c_api.cc")
.linePatterns(
"// Helper functions -+",
"// Shape functions -+",
"static TF_OperationDescription\\* TF_NewOperationLocked\\(TF_Graph\\* graph,",
"\\}",
"static TF_Operation\\* TF_FinishOperationLocked\\(TF_OperationDescription\\* desc,",
"\\}"))
.put(new Info("OutputTensor", "TensorId", "tensorflow::AttrValue").skip())
.put(
new Info("c_api_experimental.h")
.linePatterns(
"typedef struct TFE_OpAttrs TFE_OpAttrs;",
"#define TFE_CUSTOM_DEVICE_VERSION 4"))
.put(
new Info("TF_CAPI_EXPORT", "TF_Bool", "TF_GUARDED_BY", "TF_MUST_USE_RESULT")
.cppTypes()
.annotations())
.put(new Info("TF_DISALLOW_COPY_AND_ASSIGN").skip())
.put(
new Info("TF_Buffer::data")
.javaText(
"public native @Const Pointer data(); public native TF_Buffer data(Pointer data);"))
.put(
new Info("TF_Status")
.pointerTypes("TF_Status")
.base("org.tensorflow.internal.c_api.AbstractTF_Status"))
.put(
new Info("TF_Buffer")
.pointerTypes("TF_Buffer")
.base("org.tensorflow.internal.c_api.AbstractTF_Buffer"))
.put(
new Info("TF_Tensor")
.pointerTypes("TF_Tensor")
.base("org.tensorflow.internal.c_api.AbstractTF_Tensor"))
.put(
new Info("TF_Session")
.pointerTypes("TF_Session")
.base("org.tensorflow.internal.c_api.AbstractTF_Session"))
.put(
new Info("TF_SessionOptions")
.pointerTypes("TF_SessionOptions")
.base("org.tensorflow.internal.c_api.AbstractTF_SessionOptions"))
.put(
new Info("TF_Graph")
.pointerTypes("TF_Graph")
.base("org.tensorflow.internal.c_api.AbstractTF_Graph")
.purify())
.put(new Info("tensorflow::Graph").javaNames("NativeGraphPointer"))
.put(
new Info("TF_Graph::graph")
.javaText("public native @MemberGetter @ByRef NativeGraphPointer graph();"))
.put(
new Info(
"TF_Graph::refiner",
"TF_Graph::mu",
"TF_Graph::sessions",
"TF_Graph::delete_requested")
.skip())
.put(
new Info("std::unordered_map<tensorflow::string,tensorflow::Node*>")
.pointerTypes("NameMap")
.define()
.javaText("public native long erase(@StdString BytePointer key);"))
.put(
new Info("TF_Function")
.pointerTypes("TF_Function")
.base("org.tensorflow.internal.c_api.AbstractTF_Function"))
.put(
new Info("TF_ImportGraphDefOptions")
.pointerTypes("TF_ImportGraphDefOptions")
.base("org.tensorflow.internal.c_api.AbstractTF_ImportGraphDefOptions"))
.put(new Info("TF_WhileParams").purify())
.put(new Info("TF_Operation").purify())
.put(
new Info("TF_Operation::node")
.javaText("public native @MemberGetter @ByRef Node node();"))
.put(
new Info("TFE_Context")
.pointerTypes("TFE_Context")
.base("org.tensorflow.internal.c_api.AbstractTFE_Context"))
.put(
new Info("TFE_ContextOptions")
.pointerTypes("TFE_ContextOptions")
.base("org.tensorflow.internal.c_api.AbstractTFE_ContextOptions"))
.put(
new Info("TFE_Context::context")
.javaText("@MemberGetter public native @ByRef EagerContext context();"))
.put(
new Info("TFE_Op")
.pointerTypes("TFE_Op")
.base("org.tensorflow.internal.c_api.AbstractTFE_Op"))
.put(
new Info("TFE_Op::operation")
.javaText("@MemberGetter public native @ByRef EagerOperation operation();"))
.put(
new Info("TFE_TensorHandle")
.pointerTypes("TFE_TensorHandle")
.base("org.tensorflow.internal.c_api.AbstractTFE_TensorHandle"))
.put(new Info("SP_Stream").cast().pointerTypes("Pointer"))
.put(
new Info(
"TF_ShapeInferenceContextDimValueKnown",
"TFE_NewTensorHandle(const tensorflow::Tensor&, TF_Status*)",
"TF_InitKernel",
"TFE_MonitoringCounterCell",
"TFE_MonitoringCounter0",
"TFE_MonitoringCounter1",
"TFE_MonitoringCounter2",
"TFE_MonitoringIntGaugeCell",
"TFE_MonitoringIntGauge0",
"TFE_MonitoringIntGauge1",
"TFE_MonitoringIntGauge2",
"TFE_MonitoringStringGaugeCell",
"TFE_MonitoringStringGauge0",
"TFE_MonitoringStringGauge1",
"TFE_MonitoringStringGauge2",
"TFE_MonitoringBoolGaugeCell",
"TFE_MonitoringBoolGauge0",
"TFE_MonitoringBoolGauge1",
"TFE_MonitoringBoolGauge2",
"TFE_MonitoringSamplerCell",
"TFE_MonitoringSampler0",
"TFE_MonitoringSampler1",
"TFE_MonitoringSampler2",
"TFE_CustomDeviceTensorHandle",
"TFE_CustomDevice")
.skip())
.put(new Info("TF_OperationDescription").pointerTypes("TF_OperationDescription").purify())
.put(new Info("tensorflow::Scope").javaNames("TF_Scope"))
.put(new Info("tensorflow::NodeBuilder").pointerTypes("NodeBuilder"))
.put(
new Info("string", "tensorflow::string")
.annotations("@StdString")
.valueTypes("BytePointer", "String")
.pointerTypes("BytePointer"))
.put(new Info("absl::Span", "tensorflow::gtl::ArraySlice").annotations("@Span"))
.put(
new Info("std::vector<tensorflow::Output>").pointerTypes("NativeOutputVector").define())
.put(new Info("tensorflow::Output").javaNames("NativeOutput"))
.put(new Info("tensorflow::Operation").javaNames("NativeOperation"))
.put(new Info("tensorflow::Status").javaNames("NativeStatus").purify())
.put(
new Info("tensorflow::int32")
.cast()
.valueTypes("int")
.pointerTypes("IntPointer", "IntBuffer", "int[]"))
.put(
new Info(
"tensorflow::CompositeOpScopes",
"tensorflow::Input",
"tensorflow::InputList",
"tensorflow::OutputHash",
"tensorflow::StackFrame",
"tensorflow::StatusGroup",
"tensorflow::internal::TF_StatusDeleter",
"tensorflow::GraphDef",
"tensorflow::Scope::graph_as_shared_ptr",
"tensorflow::Scope::ToGraphDef",
"tensorflow::Scope::ToGraph",
"tensorflow::Scope::DoShapeInference",
"tensorflow::Scope::DisabledShapeInferenceScope",
"tensorflow::Scope::control_deps",
"tensorflow::Scope::WithKernelLabel",
"tensorflow::Scope::ClearColocation",
"tensorflow::Scope::ColocateWith",
"tensorflow::Scope::ColocateWith",
"tensorflow::Scope::WithXlaCluster",
"tensorflow::Scope::WithAssignedDevice",
"tensorflow::Scope::status",
"tensorflow::Scope::UpdateStatus",
"tensorflow::Status::code",
"tensorflow::CreateOutputWithScope",
"TF_OperationDescription::colocation_constraints",
"tensorflow::Operation::num_inputs",
"tensorflow::Operation::input_type",
"tensorflow::Operation::input",
"tensorflow::Operation::num_outputs",
"tensorflow::Operation::output_type",
"tensorflow::Operation::output",
"tensorflow::Operation::hash",
"tensorflow::Output::hash",
"tensorflow::Output::type",
"tensorflow::Status::GetAllPayloads",
"tensorflow::Status::ReplaceAllPayloads",
"tensorflow::Status::ErasePayload",
"tensorflow::Status::SetPayload",
"tensorflow::Status::GetPayload",
"tensorflow::Status::ForEachPayload",
"tensorflow::Node::SetStackTrace",
"tensorflow::Node::GetStackTrace")
.skip());
}