org.bytedeco.javacpp.presets.tensorflow Maven / Gradle / Ivy
/*
* Copyright (C) 2015-2018 Samuel Audet
*
* Licensed either under the Apache License, Version 2.0, or (at your option)
* under the terms of the GNU General Public License as published by
* the Free Software Foundation (subject to the "Classpath" exception),
* either version 2, or any later version (collectively, the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.gnu.org/licenses/
* http://www.gnu.org/software/classpath/license.html
*
* or as provided in the LICENSE.txt file that accompanied this code.
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.bytedeco.javacpp.presets;
import org.bytedeco.javacpp.BytePointer;
import org.bytedeco.javacpp.FunctionPointer;
import org.bytedeco.javacpp.Loader;
import org.bytedeco.javacpp.Pointer;
import org.bytedeco.javacpp.annotation.Adapter;
import org.bytedeco.javacpp.annotation.ByVal;
import org.bytedeco.javacpp.annotation.Cast;
import org.bytedeco.javacpp.annotation.Platform;
import org.bytedeco.javacpp.annotation.Properties;
import org.bytedeco.javacpp.annotation.StdString;
import org.bytedeco.javacpp.tools.Info;
import org.bytedeco.javacpp.tools.InfoMap;
import org.bytedeco.javacpp.tools.InfoMapper;
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
*
* @author Samuel Audet
*/
@Properties(value = {
@Platform(
value = {"linux-x86", "macosx"},
compiler = "cpp11",
define = {"NDEBUG", "UNIQUE_PTR_NAMESPACE std", "SHARED_PTR_NAMESPACE std"},
include = {
"google/protobuf/message_lite.h",
"tensorflow/core/platform/default/integral_types.h",
"tensorflow/core/lib/bfloat16/bfloat16.h",
"tensorflow/core/framework/numeric_types.h",
"tensorflow/core/platform/init_main.h",
"tensorflow/core/platform/types.h",
"tensorflow/core/platform/mutex.h",
"tensorflow/core/platform/macros.h",
"tensorflow/core/util/port.h",
"tensorflow/core/lib/core/error_codes.pb.h",
"tensorflow/core/platform/logging.h",
"tensorflow/core/lib/core/status.h",
"tensorflow/core/platform/protobuf.h",
"tensorflow/core/platform/file_system.h",
"tensorflow/core/platform/file_statistics.h",
"tensorflow/core/platform/env.h",
// "tensorflow/core/graph/dot.h",
"tensorflow/core/protobuf/debug.pb.h",
"tensorflow/core/protobuf/cluster.pb.h",
"tensorflow/core/protobuf/rewriter_config.pb.h",
"tensorflow/core/protobuf/config.pb.h",
"tensorflow/core/framework/cost_graph.pb.h",
"tensorflow/core/framework/step_stats.pb.h",
"tensorflow/core/framework/versions.pb.h",
"tensorflow/core/public/session_options.h",
"tensorflow/core/lib/core/threadpool.h",
"tensorflow/core/framework/allocation_description.pb.h",
"tensorflow/core/framework/allocator.h",
"tensorflow/core/framework/tensor_shape.pb.h",
"tensorflow/core/framework/types.pb.h",
"tensorflow/core/framework/resource_handle.pb.h",
"tensorflow/core/framework/tensor.pb.h",
"tensorflow/core/framework/tensor_description.pb.h",
"tensorflow/core/framework/tensor_types.h",
"tensorflow/core/framework/tensor_shape.h",
// "tensorflow/core/framework/tensor_slice.h",
"tensorflow/core/framework/tensor_util.h",
"tensorflow/core/framework/tensor_reference.h",
"tensorflow/core/framework/tensor.h",
"tensorflow/core/framework/attr_value.pb.h",
"tensorflow/core/framework/node_def.pb.h",
"tensorflow/core/framework/op_def.pb.h",
"tensorflow/core/framework/function.pb.h",
"tensorflow/core/framework/graph.pb.h",
"tensorflow/core/framework/session_state.h",
"tensorflow/core/framework/types.h",
"tensorflow/core/framework/control_flow.h",
"tensorflow/core/framework/kernel_def_builder.h",
"tensorflow/core/framework/tracking_allocator.h",
"tensorflow/core/framework/op_kernel.h",
"tensorflow/core/framework/op_segment.h",
"tensorflow/core/framework/shape_inference.h",
"tensorflow/core/framework/partial_tensor_shape.h",
"tensorflow/core/framework/device_attributes.pb.h",
"tensorflow/core/public/session.h",
"tensorflow/core/framework/tensor_slice.pb.h",
"tensorflow/core/framework/tensor_slice.h",
"tensorflow/core/util/tensor_slice_set.h",
"tensorflow/core/util/tensor_slice_util.h",
"tensorflow/core/util/tensor_slice_reader.h",
"tensorflow/core/util/tensor_bundle/tensor_bundle.h",
"tensorflow/c/tf_status_helper.h",
"tensorflow/c/checkpoint_reader.h",
"tensorflow/c/c_api.h",
"tensorflow/core/framework/op_def.pb.h",
"tensorflow/core/framework/op_def_builder.h",
"tensorflow/core/framework/op_def_util.h",
"tensorflow/core/framework/op.h",
"tensorflow/core/framework/types.h",
"tensorflow/core/graph/edgeset.h",
"tensorflow/core/lib/gtl/iterator_range.h",
// "tensorflow/core/lib/gtl/inlined_vector.h",
"tensorflow/core/framework/function.h",
"tensorflow/core/util/device_name_utils.h",
"tensorflow/core/framework/device_attributes.pb.h",
"tensorflow/core/framework/device_base.h",
"tensorflow/core/common_runtime/device.h",
"tensorflow/core/common_runtime/device_mgr.h",
"tensorflow/core/common_runtime/process_function_library_runtime.h",
"tensorflow/core/graph/graph.h",
"tensorflow/core/graph/tensor_id.h",
"tensorflow/core/framework/node_def_builder.h",
"tensorflow/core/framework/node_def_util.h",
"tensorflow/core/framework/selective_registration.h",
"tensorflow/core/graph/node_builder.h",
"tensorflow/core/graph/graph_def_builder.h",
"tensorflow/core/graph/default_device.h",
"tensorflow/core/graph/graph_constructor.h",
"tensorflow/core/graph/gradients.h",
"tensorflow/cc/framework/scope.h",
"tensorflow/cc/framework/ops.h",
"tensorflow/core/framework/api_def.pb.h",
"tensorflow/core/framework/op_gen_lib.h",
"tensorflow/cc/framework/cc_op_gen.h",
"tensorflow/cc/framework/gradients.h",
"tensorflow/core/protobuf/saver.pb.h",
"tensorflow/core/protobuf/meta_graph.pb.h",
"tensorflow/cc/saved_model/loader.h",
"tensorflow_adapters.h",
"tensorflow/cc/ops/standard_ops.h",
"tensorflow/cc/ops/const_op.h",
"tensorflow/cc/ops/array_ops.h",
"tensorflow/cc/ops/candidate_sampling_ops.h",
"tensorflow/cc/ops/control_flow_ops.h",
"tensorflow/cc/ops/data_flow_ops.h",
"tensorflow/cc/ops/image_ops.h",
"tensorflow/cc/ops/io_ops.h",
"tensorflow/cc/ops/linalg_ops.h",
"tensorflow/cc/ops/logging_ops.h",
"tensorflow/cc/ops/math_ops.h",
"tensorflow/cc/ops/nn_ops.h",
"tensorflow/cc/ops/no_op.h",
"tensorflow/cc/ops/parsing_ops.h",
"tensorflow/cc/ops/random_ops.h",
"tensorflow/cc/ops/sparse_ops.h",
"tensorflow/cc/ops/state_ops.h",
"tensorflow/cc/ops/string_ops.h",
"tensorflow/cc/ops/training_ops.h",
"tensorflow/cc/ops/user_ops.h"},
link = "tensorflow_cc", preload = "tensorflow_framework"),
@Platform(value = {"linux-x86_64", "macosx-x86_64"}, extension = "-gpu"),
@Platform(value = {"android"},
compiler = {"cpp11"},
define = {"NDEBUG", "UNIQUE_PTR_NAMESPACE std"},
include = {
"google/protobuf/message_lite.h",
"tensorflow/core/platform/default/integral_types.h",
"tensorflow/core/lib/bfloat16/bfloat16.h",
"tensorflow/core/framework/numeric_types.h",
"tensorflow/core/platform/init_main.h",
"tensorflow/core/platform/types.h",
"tensorflow/core/platform/mutex.h",
"tensorflow/core/platform/macros.h",
"tensorflow/core/util/port.h",
"tensorflow/core/lib/core/error_codes.pb.h",
"tensorflow/core/platform/logging.h",
"tensorflow/core/lib/core/status.h",
"tensorflow/core/platform/protobuf.h",
"tensorflow/core/platform/file_system.h",
"tensorflow/core/platform/file_statistics.h",
"tensorflow/core/platform/env.h",
"tensorflow/core/protobuf/debug.pb.h",
"tensorflow/core/protobuf/cluster.pb.h",
"tensorflow/core/protobuf/rewriter_config.pb.h",
"tensorflow/core/protobuf/config.pb.h",
"tensorflow/core/framework/cost_graph.pb.h",
"tensorflow/core/framework/step_stats.pb.h",
"tensorflow/core/framework/versions.pb.h",
"tensorflow/core/public/session_options.h",
"tensorflow/core/lib/core/threadpool.h",
"tensorflow/core/framework/allocation_description.pb.h",
"tensorflow/core/framework/allocator.h",
"tensorflow/core/framework/tensor_shape.pb.h",
"tensorflow/core/framework/types.pb.h",
"tensorflow/core/framework/resource_handle.pb.h",
"tensorflow/core/framework/tensor.pb.h",
"tensorflow/core/framework/tensor_description.pb.h",
"tensorflow/core/framework/tensor_types.h",
"tensorflow/core/framework/tensor_shape.h",
"tensorflow/core/framework/tensor_util.h",
"tensorflow/core/framework/tensor_reference.h",
"tensorflow/core/framework/tensor.h",
"tensorflow/core/framework/attr_value.pb.h",
"tensorflow/core/framework/node_def.pb.h",
"tensorflow/core/framework/function.pb.h",
"tensorflow/core/framework/graph.pb.h",
"tensorflow/core/framework/session_state.h",
"tensorflow/core/framework/types.h",
"tensorflow/core/framework/control_flow.h",
"tensorflow/core/framework/kernel_def_builder.h",
"tensorflow/core/framework/tracking_allocator.h",
"tensorflow/core/framework/op_kernel.h",
"tensorflow/core/framework/op_segment.h",
"tensorflow/core/framework/shape_inference.h",
"tensorflow/core/framework/partial_tensor_shape.h",
"tensorflow/core/framework/device_attributes.pb.h",
"tensorflow/core/public/session.h",
"tensorflow/core/framework/tensor_slice.pb.h",
"tensorflow/core/framework/tensor_slice.h",
"tensorflow/core/util/tensor_slice_set.h",
"tensorflow/core/util/tensor_slice_util.h",
"tensorflow/core/util/tensor_slice_reader.h",
"tensorflow/core/util/tensor_bundle/tensor_bundle.h",
"tensorflow/c/tf_status_helper.h",
"tensorflow/c/checkpoint_reader.h",
"tensorflow/c/c_api.h",
"tensorflow/core/framework/op_def.pb.h",
"tensorflow/core/framework/op_def_builder.h",
"tensorflow/core/framework/op_def_util.h",
"tensorflow/core/framework/op.h",
"tensorflow/core/framework/types.h",
"tensorflow/core/graph/edgeset.h",
"tensorflow/core/lib/gtl/iterator_range.h",
"tensorflow/core/framework/function.h",
"tensorflow/core/util/device_name_utils.h",
"tensorflow/core/framework/device_attributes.pb.h",
"tensorflow/core/framework/device_base.h",
"tensorflow/core/common_runtime/device.h",
"tensorflow/core/common_runtime/device_mgr.h",
"tensorflow/core/common_runtime/process_function_library_runtime.h",
"tensorflow/core/graph/graph.h",
"tensorflow/core/graph/tensor_id.h",
"tensorflow/core/framework/node_def_builder.h",
"tensorflow/core/framework/node_def_util.h",
"tensorflow/core/graph/node_builder.h",
"tensorflow/core/graph/graph_def_builder.h",
"tensorflow/core/graph/default_device.h",
"tensorflow/core/graph/graph_constructor.h",
"tensorflow/core/protobuf/saver.pb.h",
"tensorflow/core/protobuf/meta_graph.pb.h",
"tensorflow_adapters.h"},
link = "tensorflow_cc", preload = "tensorflow_framework"),
},
target = "org.bytedeco.javacpp.tensorflow",
helper = "org.bytedeco.javacpp.helper.tensorflow")
public class tensorflow implements InfoMapper {
public void map(InfoMap infoMap) {
infoMap.put(new Info("tensorflow_adapters.h").skip())
.put(new Info("B16_DEVICE_FUNC", "EIGEN_ALWAYS_INLINE", "EIGEN_DEVICE_FUNC", "EIGEN_STRONG_INLINE", "PROTOBUF_CONSTEXPR", "PROTOBUF_FINAL",
"TF_FALLTHROUGH_INTENDED", "TF_ATTRIBUTE_NORETURN", "TF_ATTRIBUTE_NOINLINE", "TF_ATTRIBUTE_UNUSED",
"TF_ATTRIBUTE_COLD", "TF_ATTRIBUTE_WEAK", "TF_PACKED", "TF_MUST_USE_RESULT", "SHOULD_REGISTER_OP_GRADIENT",
"TF_EXPORT", "TF_ATTRIBUTE_ALWAYS_INLINE").cppTypes().annotations())
.put(new Info("TF_CHECK_OK", "TF_QCHECK_OK").cppTypes("void", "tensorflow::Status"))
.put(new Info("TF_DISALLOW_COPY_AND_ASSIGN").cppText("#define TF_DISALLOW_COPY_AND_ASSIGN(TypeName)"))
.put(new Info("GOOGLE_PROTOBUF_DEPRECATED_ATTR", "PROTOBUF_DEPRECATED_ATTR").cppTypes().annotations("@Deprecated"))
.put(new Info("SWIG", "TENSORFLOW_LITE_PROTOS").define(true))
.put(new Info("TENSORFLOW_USE_SYCL").define(false))
.put(new Info("std::hash").pointerTypes("HalfHash"))
.put(new Info("Eigen::NumTraits").pointerTypes("bfloat16NumTraits"))
.put(new Info("Eigen::half").cast().valueTypes("short").pointerTypes("ShortPointer", "ShortBuffer", "short..."))
.put(new Info("short", "tensorflow::int16", "tensorflow::uint16").valueTypes("short").pointerTypes("ShortPointer", "ShortBuffer", "short..."))
.put(new Info("int", "int32", "tensorflow::int32", "tensorflow::uint32").valueTypes("int").pointerTypes("IntPointer", "IntBuffer", "int..."))
.put(new Info("long long", "tensorflow::int64", "tensorflow::uint64", "std::size_t",
"tensorflow::Microseconds", "tensorflow::Nanoseconds", "tensorflow::Bytes").cast().valueTypes("long").pointerTypes("LongPointer", "LongBuffer", "long..."))
.put(new Info("float").valueTypes("float").pointerTypes("FloatPointer", "FloatBuffer", "float..."))
.put(new Info("double").valueTypes("double").pointerTypes("DoublePointer", "DoubleBuffer", "double..."))
.put(new Info("bool").cast().valueTypes("boolean").pointerTypes("BoolPointer", "boolean..."))
.put(new Info("std::complex").cast().pointerTypes("FloatPointer", "FloatBuffer", "float..."))
.put(new Info("std::initializer_list").skip())
.put(new Info("string", "std::string", "tensorflow::string").annotations("@StdString").valueTypes("BytePointer", "String").pointerTypes("@Cast({\"char*\", \"std::string*\"}) BytePointer"))
.put(new Info("std::unordered_set").pointerTypes("StringSet").define())
.put(new Info("std::vector").pointerTypes("StringPieceVector").define())
.put(new Info("std::vector", "std::vector").pointerTypes("StringVector").define())
.put(new Info("std::vector >").pointerTypes("StringStringPairVector").define())
.put(new Info("std::condition_variable", "std::mutex", "std::unique_lock",
"tensorflow::condition_variable", "tensorflow::mutex", "tensorflow::mutex_lock").cast().pointerTypes("Pointer"))
.put(new Info("google::protobuf::int8", "google::protobuf::uint8").cast().valueTypes("byte").pointerTypes("BytePointer", "ByteBuffer", "byte[]"))
.put(new Info("google::protobuf::int16", "google::protobuf::uint16").cast().valueTypes("short").pointerTypes("ShortPointer", "ShortBuffer", "short[]"))
.put(new Info("google::protobuf::int32", "google::protobuf::uint32").cast().valueTypes("int").pointerTypes("IntPointer", "IntBuffer", "int[]"))
.put(new Info("google::protobuf::int64", "google::protobuf::uint64").cast().valueTypes("long").pointerTypes("LongPointer", "LongBuffer", "long[]"))
.put(new Info("google::protobuf::Message").cast().pointerTypes("MessageLite"))
.put(new Info("google::protobuf::Any", "google::protobuf::Descriptor", "google::protobuf::EnumDescriptor", "google::protobuf::Metadata").cast().pointerTypes("Pointer"))
.put(new Info("google::protobuf::Map", "google::protobuf::RepeatedField", "google::protobuf::RepeatedPtrField", "protobuf::RepeatedPtrField",
"google::protobuf::internal::ExplicitlyConstructed", "google::protobuf::internal::MapEntry", "google::protobuf::internal::MapField",
"google::protobuf::internal::AuxillaryParseTableField", "google::protobuf::internal::ParseTableField", "google::protobuf::internal::ParseTable",
"google::protobuf::internal::FieldMetadata", "google::protobuf::internal::SerializationTable", "google::protobuf::internal::proto3_preserve_unknown_",
"google::protobuf::is_proto_enum", "google::protobuf::GetEnumDescriptor").skip())
.put(new Info("tensorflow::error::protobuf_tensorflow_2fcore_2flib_2fcore_2ferror_5fcodes_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fprotobuf_2fdebug_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fprotobuf_2fconfig_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fcost_5fgraph_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fstep_5fstats_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fversions_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fallocation_5fdescription_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2ftensor_5fshape_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2ftypes_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fresource_5fhandle_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2ftensor_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2ftensor_5fdescription_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fattr_5fvalue_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fnode_5fdef_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fop_5fdef_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2ffunction_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fgraph_2eproto::TableStruct",
"tensorflow::protobuf_tensorflow_2fcore_2fframework_2fdevice_5fattributes_2eproto::TableStruct", "tensorflow::VariantTensorDataProtoDefaultTypeInternal",
"tensorflow::JobDef_TasksEntryDefaultTypeInternal", "tensorflow::ResourceHandleProtoDefaultTypeInternal", "tensorflow::NameAttrList_AttrEntryDefaultTypeInternal",
"tensorflow::NodeDef_AttrEntryDefaultTypeInternal", "tensorflow::FunctionDef_AttrEntryDefaultTypeInternal", "tensorflow::FunctionDef_RetEntryDefaultTypeInternal",
"tensorflow::DeviceAttributesDefaultTypeInternal", "tensorflow::DeviceLocalityDefaultTypeInternal", "tensorflow::ConfigProto_DeviceCountEntryDefaultTypeInternal",
"tensorflow::AutoParallelOptionsDefaultTypeInternal", "tensorflow::ClusterDefDefaultTypeInternal", "tensorflow::JobDefDefaultTypeInternal",
"tensorflow::DebugOptionsDefaultTypeInternal", "tensorflow::DebugTensorWatchDefaultTypeInternal", "tensorflow::AllocatorMemoryUsedDefaultTypeInternal",
"tensorflow::ConfigProtoDefaultTypeInternal", "tensorflow::CostGraphDefDefaultTypeInternal", "tensorflow::CostGraphDef_NodeDefaultTypeInternal",
"tensorflow::CostGraphDef_Node_InputInfoDefaultTypeInternal", "tensorflow::CostGraphDef_Node_OutputInfoDefaultTypeInternal", "tensorflow::DeviceStepStatsDefaultTypeInternal",
"tensorflow::GPUOptionsDefaultTypeInternal", "tensorflow::GraphDefDefaultTypeInternal", "tensorflow::GraphOptionsDefaultTypeInternal", "tensorflow::MemoryStatsDefaultTypeInternal",
"tensorflow::NodeExecStatsDefaultTypeInternal", "tensorflow::NodeOutputDefaultTypeInternal", "tensorflow::OptimizerOptionsDefaultTypeInternal",
"tensorflow::RPCOptionsDefaultTypeInternal", "tensorflow::RewriterConfigDefaultTypeInternal", "tensorflow::RunMetadataDefaultTypeInternal",
"tensorflow::RunOptionsDefaultTypeInternal", "tensorflow::StepStatsDefaultTypeInternal", "tensorflow::ThreadPoolOptionProtoDefaultTypeInternal",
"tensorflow::TensorShapeProtoDefaultTypeInternal", "tensorflow::TensorShapeProto_DimDefaultTypeInternal", "tensorflow::AllocationDescriptionDefaultTypeInternal",
"tensorflow::TensorDescriptionDefaultTypeInternal", "tensorflow::VersionDefDefaultTypeInternal", "tensorflow::ResourceHandleDefaultTypeInternal",
"tensorflow::TensorProtoDefaultTypeInternal", "tensorflow::AttrValueDefaultTypeInternal", "tensorflow::AttrValue_ListValueDefaultTypeInternal",
"tensorflow::NameAttrListDefaultTypeInternal", "tensorflow::NodeDefDefaultTypeInternal", "tensorflow::OpDefDefaultTypeInternal", "tensorflow::OpDef_ArgDefDefaultTypeInternal",
"tensorflow::OpDef_AttrDefDefaultTypeInternal", "tensorflow::OpDeprecationDefaultTypeInternal", "tensorflow::OpListDefaultTypeInternal",
"tensorflow::FunctionDefDefaultTypeInternal", "tensorflow::FunctionDefLibraryDefaultTypeInternal", "tensorflow::GradientDefDefaultTypeInternal",
"tensorflow::SaverDefDefaultTypeInternal", "tensorflow::AssetFileDefDefaultTypeInternal", "tensorflow::CollectionDefDefaultTypeInternal",
"tensorflow::CollectionDef_AnyListDefaultTypeInternal", "tensorflow::CollectionDef_BytesListDefaultTypeInternal", "tensorflow::CollectionDef_FloatListDefaultTypeInternal",
"tensorflow::CollectionDef_Int64ListDefaultTypeInternal", "tensorflow::CollectionDef_NodeListDefaultTypeInternal", "tensorflow::MetaGraphDefDefaultTypeInternal",
"tensorflow::MetaGraphDef_CollectionDefEntryDefaultTypeInternal", "tensorflow::MetaGraphDef_CollectionDefEntryDefaultTypeInternal",
"tensorflow::MetaGraphDef_MetaInfoDefDefaultTypeInternal", "tensorflow::MetaGraphDef_SignatureDefEntryDefaultTypeInternal",
"tensorflow::MetaGraphDef_SignatureDefEntryDefaultTypeInternal", "tensorflow::SignatureDefDefaultTypeInternal", "tensorflow::SignatureDef_InputsEntryDefaultTypeInternal",
"tensorflow::SignatureDef_OutputsEntryDefaultTypeInternal", "tensorflow::TensorInfoDefaultTypeInternal", "tensorflow::TensorInfo_CooSparseDefaultTypeInternal",
"tensorflow::TensorSliceProtoDefaultTypeInternal", "tensorflow::TensorSliceProto_ExtentDefaultTypeInternal", "tensorflow::ApiDefDefaultTypeInternal",
"tensorflow::ApiDef_ArgDefaultTypeInternal", "tensorflow::ApiDef_AttrDefaultTypeInternal", "tensorflow::ApiDef_EndpointDefaultTypeInternal",
"tensorflow::ApiDefsDefaultTypeInternal", "tensorflow::DebuggedSourceFileDefaultTypeInternal", "tensorflow::DebuggedSourceFilesDefaultTypeInternal",
"tensorflow::AllocationRecordDefaultTypeInternal","tensorflow::GPUOptions_ExperimentalDefaultTypeInternal", "tensorflow::GPUOptions_Experimental_VirtualDevicesDefaultTypeInternal",
"tensorflow::InterconnectLinkDefaultTypeInternal", "tensorflow::LocalLinksDefaultTypeInternal",
"tensorflow::JobDef_TasksEntry_DoNotUseDefaultTypeInternal", "tensorflow::ConfigProto_DeviceCountEntry_DoNotUseDefaultTypeInternal",
"tensorflow::NameAttrList_AttrEntry_DoNotUseDefaultTypeInternal", "tensorflow::NodeDef_AttrEntry_DoNotUseDefaultTypeInternal",
"tensorflow::FunctionDef_AttrEntry_DoNotUseDefaultTypeInternal", "tensorflow::FunctionDef_RetEntry_DoNotUseDefaultTypeInternal",
"tensorflow::MetaGraphDef_CollectionDefEntry_DoNotUseDefaultTypeInternal", "tensorflow::MetaGraphDef_SignatureDefEntry_DoNotUseDefaultTypeInternal",
"tensorflow::SignatureDef_InputsEntry_DoNotUseDefaultTypeInternal", "tensorflow::SignatureDef_OutputsEntry_DoNotUseDefaultTypeInternal").skip())
.put(new Info("tensorflow::core::RefCounted").cast().pointerTypes("Pointer"))
.put(new Info("tensorflow::ConditionResult").cast().valueTypes("int"))
.put(new Info("tensorflow::protobuf::Message", "tensorflow::protobuf::MessageLite").cast().pointerTypes("MessageLite"))
.put(new Info("tensorflow::Allocator::is_simple").skip())
.put(new Info("basic/containers").cppTypes("tensorflow::gtl::InlinedVector", "google::protobuf::Map", "tensorflow::gtl::FlatMap"))
.put(new Info("tensorflow::TrackingAllocator").purify())
.put(new Info("tensorflow::DeviceContext").pointerTypes("DeviceContext"))
.put(new Info("tensorflow::register_kernel::Name").pointerTypes("RegisterKernelName"))
.put(new Info("tensorflow::register_kernel::system::Name").pointerTypes("RegisterKernelSystemName"))
.put(new Info("tensorflow::DataType").cast().valueTypes("int").pointerTypes("IntPointer"))
.put(new Info("std::pair").pointerTypes("WrappedAllocator").define())
.put(new Info("std::tuple").cast().pointerTypes("SizeTPointer"))
.put(new Info("std::vector").pointerTypes("DeviceVector").define())
.put(new Info("std::vector").pointerTypes("DeviceContextVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("AllocatorAttributesVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("AllocRecordVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("DeviceContextInlinedVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("DeviceTypeVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("TensorValueVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("WrappedAllocatorVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("LongVector").define())
.put(new Info("tensorflow::gtl::InlinedVector").pointerTypes("DataTypeVector").define())
.put(new Info("tensorflow::DataTypeSlice").cast().pointerTypes("DataTypeVector"))
.put(new Info("tensorflow::NumberTypes", "tensorflow::QuantizedTypes", "tensorflow::RealAndQuantizedTypes").skip())
.put(new Info("tensorflow::OpArgIterator").pointerTypes("MutableTensorOpArgIterator").define())
.put(new Info("tensorflow::OpArgIterator::operator *()").skip())
.put(new Info("tensorflow::OpArgIterator").pointerTypes("TensorOpArgIterator").define())
.put(new Info("tensorflow::OpArgIterator::operator *()").skip())
.put(new Info("tensorflow::Tensor").base("AbstractTensor"))
.put(new Info("tensorflow::Session").base("AbstractSession"))
.put(new Info("tensorflow::Session::~Session()").javaText("/** Calls {@link tensorflow#NewSession(SessionOptions)} and registers a deallocator. */\n"
+ "public Session(SessionOptions options) { super(options); }"))
.put(new Info("tensorflow::TensorShapeBase", "tensorflow::TensorShapeBase").pointerTypes("TensorShapeBase"))
.put(new Info("tensorflow::TensorShapeIter").pointerTypes("TensorShapeIter").define())
.put(new Info("tensorflow::shape_inference::InferenceContext").purify())
.put(new Info("std::vector > >",
"std::vector > > >",
"std::vector > > >").skip())
.put(new Info("std::pair").pointerTypes("ShapeHandlePair").define())
.put(new Info("std::pair").pointerTypes("DimensionHandlePair").define())
.put(new Info("std::vector").pointerTypes("TensorVector").define())
.put(new Info("std::vector").pointerTypes("TensorProtoVector").define())
.put(new Info("std::vector").pointerTypes("TensorShapeVector").define())
.put(new Info("std::vector").pointerTypes("NodeOutVector").define())
.put(new Info("std::vector").pointerTypes("NodeVector").define())
.put(new Info("std::vector >").pointerTypes("NodeIntPairVector").define())
.put(new Info("google::protobuf::Map::const_iterator", "AttrValueMap::const_iterator").skip())
.put(new Info("google::protobuf::Map",
"tensorflow::protobuf::Map").pointerTypes("StringAttrValueMap").define())
.put(new Info("tensorflow::FunctionDefHelper::AttrValueWrapper").pointerTypes("FunctionDefHelper.AttrValueWrapper"))
.put(new Info("std::vector >",
"tensorflow::gtl::ArraySlice >").cast().pointerTypes("StringAttrPairVector").define())
.put(new Info("tensorflow::ops::NodeOut").valueTypes("@ByVal NodeBuilder.NodeOut", "Node"))
.put(new Info("tensorflow::NodeBuilder::NodeOut").pointerTypes("NodeBuilder.NodeOut"))
.put(new Info("std::function)>").cast().pointerTypes("Pointer"))
.put(new Info("std::vector::iterator").skip())
.put(new Info("std::vector::const_iterator").skip())
.put(new Info("tensorflow::ops::Cast").cppTypes("class tensorflow::ops::Cast").pointerTypes("CastOp"))
.put(new Info("tensorflow::ops::Const").cppTypes("class tensorflow::ops::Const").pointerTypes("ConstOp"))
.put(new Info("mode_t").skip())
.put(new Info("tensorflow::gtl::ArraySlice").cast().pointerTypes("StringPieceVector"))
.put(new Info("tensorflow::gtl::ArraySlice", "tensorflow::gtl::ArraySlice").cast().pointerTypes("StringVector"))
.put(new Info("tensorflow::gtl::ArraySlice >").cast().pointerTypes("StringStringPairVector"))
.put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("TensorVector"))
.put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("TensorProtoVector"))
.put(new Info("tensorflow::gtl::ArraySlice").cast().pointerTypes("TensorShapeVector"))
.put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("NodeOutVector"))
.put(new Info("tensorflow::gtl::ArraySlice")/*.cast()*/.pointerTypes("NodeVector"))
.put(new Info("tensorflow::gtl::iterator_range").pointerTypes("NeighborIterRange").define())
.put(new Info("tensorflow::gtl::iterator_range()").skip())
.put(new Info("tensorflow::gtl::iterator_range").pointerTypes("NodeIterRange").define())
.put(new Info("tensorflow::gtl::iterator_range()").skip())
// .put(new Info("std::unordered_map >").pointerTypes("NameRangeMap").define())
.put(new Info("tensorflow::gtl::FlatMap,tensorflow::hash >").pointerTypes("NameRangeMap").define())
// Skip composite op scopes bc: call to implicitly-deleted default constructor of '::tensorflow::CompositeOpScopes'
.put(new Info("tensorflow::CompositeOpScopes").skip())
// Fixed shape inference
.put(new Info("std::vector").pointerTypes("ConstTensorPtrVector").define())
.put(new Info("std::vector").pointerTypes("ConstDimensionPtrVector").define())
.put(new Info("std::vector >",
"std::vector >").pointerTypes("StringTensorPairVector").define())
.put(new Info("std::vector", "std::vector").cast().pointerTypes("EdgeVector").define())
.put(new Info("std::pair").pointerTypes("EdgeSetBoolPair").define())
.put(new Info("tensorflow::EdgeSet::const_iterator", "tensorflow::EdgeSet::iterator").pointerTypes("EdgeSetIterator"))
.put(new Info("tensorflow::GraphEdgesIterable::const_iterator").purify())
.put(new Info("tensorflow::register_op::OpDefBuilderWrapper").pointerTypes("TrueOpDefBuilderWrapper"))
.put(new Info("tensorflow::register_op::OpDefBuilderWrapper").pointerTypes("FalseOpDefBuilderWrapper"))
.put(new Info("tensorflow::checkpoint::TensorSliceSet::SliceInfo").pointerTypes("TensorSliceSet.SliceInfo"))
.put(new Info("std::pair").pointerTypes("StringPieceIntPair").define())
.put(new Info("std::pair").pointerTypes("TensorSlideStringPair").define())
.put(new Info("std::map").pointerTypes("TensorIdTensorIdMap").define())
.put(new Info("std::unordered_map").pointerTypes("VarToShapeMap").define())
.put(new Info("std::unordered_map").pointerTypes("VarToDataTypeMap").define())
.put(new Info("std::unordered_map").pointerTypes("StringTensorSliceSetMap").define())
.put(new Info("const std::unordered_map").pointerTypes("StringSliceInfoMap").define())
.put(new Info("std::vector::iterator", "std::vector::const_iterator").skip())
.put(new Info("tensorflow::ImportGraphDefResults::Index").cast().valueTypes("int").pointerTypes("IntPointer"))
.put(new Info("std::pair").pointerTypes("NodeIndexPair").define())
.put(new Info("TF_WhileParams").purify())
.put(new Info("TF_LoadSessionFromSavedModel").annotations("@Platform(not=\"android\")").javaNames("TF_LoadSessionFromSavedModel"))
.put(new Info("TF_GraphImportGraphDefOptionsRemapControlDependency").annotations("@Platform(not=\"android\")").javaNames("TF_GraphImportGraphDefOptionsRemapControlDependency"))
.put(new Info("tensorflow::SavedModelBundle::session").javaText("public native @MemberGetter @UniquePtr Session session();"))
.put(new Info("std::function").pointerTypes("Fn"))
.put(new Info("std::function").pointerTypes("ForFn"))
.put(new Info("std::function").pointerTypes("ParallelForFn"))
.put(new Info("std::function").pointerTypes("FactoryFn"))
.put(new Info("tensorflow::OpRegistrationData::shape_inference_fn")
.javaText("@MemberSetter public native OpRegistrationData shape_inference_fn(@ByVal ShapeInferenceFn shape_inference_fn);"))
.put(new Info("tensorflow::shape_inference::InferenceContext::Run")
.javaText("public native @ByVal Status Run(@ByVal ShapeInferenceFn fn);"))
.put(new Info("tensorflow::ConstantFoldingOptions::consider")
.javaText("@MemberSetter public native ConstantFoldingOptions consider(@ByVal ConsiderFunction consider);"))
.put(new Info("tensorflow::GraphConstructorOptions::cse_consider_function")
.javaText("@MemberSetter public native GraphConstructorOptions cse_consider_function(@ByVal ConsiderFunction cse_consider_function);"));
String[] attrs = {"int", "long long", "float", "double", "bool", "std::string",
"tensorflow::Tensor", "tensorflow::TensorProto", "tensorflow::TensorShape",
"tensorflow::NameAttrList", "tensorflow::StringPiece"};
for (int i = 0; i < attrs.length; i++) {
infoMap.put(new Info("tensorflow::GraphDefBuilder::Options::WithAttr<" + attrs[i] + ">").javaNames("WithAttr"));
if (i < attrs.length - 2) {
infoMap.put(new Info("tensorflow::GraphDefBuilder::Options::WithAttr >").javaNames("WithAttr"));
}
}
infoMap.put(new Info("tensorflow::DotOptions::edge_label")
.javaText("@MemberSetter public native DotOptions edge_label(EdgeLabelFunction edge_label_function);"))
.put(new Info("tensorflow::DotOptions::node_label")
.javaText("@MemberSetter public native DotOptions node_label(NodeLabelFunction node_label_function);"))
.put(new Info("tensorflow::DotOptions::edge_cost")
.javaText("@MemberSetter public native DotOptions edge_cost(EdgeCostFunction edge_cost_function);"))
.put(new Info("tensorflow::DotOptions::node_cost")
.javaText("@MemberSetter public native DotOptions node_cost(NodeCostFunction node_cost_function);"))
.put(new Info("tensorflow::DotOptions::node_color")
.javaText("@MemberSetter public native DotOptions node_color(NodeColorFunction node_color_function);"))
.put(new Info("std::function").pointerTypes("EdgeCostFunction"))
.put(new Info("std::function").pointerTypes("NodeCostFunction"))
.put(new Info("std::function").pointerTypes("NodeLabelFunction"))
.put(new Info("std::function").pointerTypes("EdgeLabelFunction"))
.put(new Info("std::function").pointerTypes("NodeColorFunction"));
infoMap.put(new Info("tensorflow::gtl::ArraySlice").annotations("@ArraySlice"))
.put(new Info("tensorflow::StringPiece").annotations("@StringPiece").valueTypes("BytePointer", "String").pointerTypes("@Cast({\"char*\", \"StringPiece*\"}) BytePointer"))
.put(new Info("tensorflow::Input::Initializer").pointerTypes("Input.Initializer").valueTypes("@Const @ByRef Input.Initializer",
"@ByRef Tensor", "byte", "short", "int", "long", "float", "double", "boolean", "@StdString String", "@StdString BytePointer"));
String[] consts = {"unsigned char", "short", "int", "long long", "float", "double", "bool", "std::string", "tensorflow::StringPiece"};
for (int i = 0; i < consts.length; i++) {
infoMap.put(new Info("tensorflow::ops::Const<" + consts[i] + ">").javaNames("Const"));
}
}
public static class Fn extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public Fn(Pointer p) { super(p); }
protected Fn() { allocate(); }
private native void allocate();
public native void call();
}
public static class ForFn extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public ForFn(Pointer p) { super(p); }
protected ForFn() { allocate(); }
private native void allocate();
public native void call(long from, long to);
}
public static class ParallelForFn extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public ParallelForFn(Pointer p) { super(p); }
protected ParallelForFn() { allocate(); }
private native void allocate();
public native int call(long from, long to, int i);
}
public static class ConsiderFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public ConsiderFunction(Pointer p) { super(p); }
protected ConsiderFunction() { allocate(); }
private native void allocate();
public native @Cast("bool") boolean call(@Cast("const tensorflow::Node*") Pointer node);
}
public static class NodeColorFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public NodeColorFunction(Pointer p) { super(p); }
protected NodeColorFunction() { allocate(); }
private native void allocate();
public native @Cast("int") int call(@Cast("const tensorflow::Node*") Pointer node);
}
public static class NodeCostFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public NodeCostFunction(Pointer p) { super(p); }
protected NodeCostFunction() { allocate(); }
private native void allocate();
public native @Cast("double") double call(@Cast("const tensorflow::Node*") Pointer node);
}
public static class EdgeCostFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public EdgeCostFunction(Pointer p) { super(p); }
protected EdgeCostFunction() { allocate(); }
private native void allocate();
public native @Cast("double") double call(@Cast("const tensorflow::Edge*") Pointer node);
}
public static class NodeLabelFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public NodeLabelFunction(Pointer p) { super(p); }
protected NodeLabelFunction() { allocate(); }
private native void allocate();
public native @StdString BytePointer call(@Cast("const tensorflow::Node*") Pointer node);
}
public static class EdgeLabelFunction extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public EdgeLabelFunction(Pointer p) { super(p); }
protected EdgeLabelFunction() { allocate(); }
private native void allocate();
public native @StdString BytePointer call(@Cast("const tensorflow::Edge*") Pointer node);
}
public static class FactoryFn extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public FactoryFn(Pointer p) { super(p); }
protected FactoryFn() { allocate(); }
private native void allocate();
public native @Cast("tensorflow::FileSystem*") Pointer call();
}
public static class ShapeInferenceFn extends FunctionPointer {
static { Loader.load(); }
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
public ShapeInferenceFn(Pointer p) { super(p); }
protected ShapeInferenceFn() { allocate(); }
private native void allocate();
public native @ByVal @Cast("tensorflow::Status*") Pointer call(@Cast("shape_inference::InferenceContext*") Pointer node);
}
@Documented @Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.PARAMETER})
@Cast({"tensorflow::gtl::ArraySlice", "&"}) @Adapter("ArraySliceAdapter")
public @interface ArraySlice { String value() default ""; }
@Documented @Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.METHOD, ElementType.PARAMETER})
@Cast("tensorflow::StringPiece&") @Adapter("StringPieceAdapter")
public @interface StringPiece { }
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy