Home | History | Annotate | Download | only in layers
      1 # Copyright 2015 The TensorFlow Authors. All Rights Reserved.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 # ==============================================================================
     15 """Ops for building neural network layers, regularizers, summaries, etc.
     16 
     17 See the @{$python/contrib.layers} guide.
     18 
     19 @@avg_pool2d
     20 @@avg_pool3d
     21 @@batch_norm
     22 @@convolution2d
     23 @@convolution3d
     24 @@conv2d_in_plane
     25 @@convolution2d_in_plane
     26 @@conv2d_transpose
     27 @@convolution2d_transpose
     28 @@conv3d_transpose
     29 @@convolution3d_transpose
     30 @@dense_to_sparse
     31 @@dropout
     32 @@elu
     33 @@embedding_lookup_unique
     34 @@flatten
     35 @@fully_connected
     36 @@GDN
     37 @@gdn
     38 @@images_to_sequence
     39 @@layer_norm
     40 @@linear
     41 @@max_pool2d
     42 @@max_pool3d
     43 @@one_hot_encoding
     44 @@relu
     45 @@relu6
     46 @@repeat
     47 @@recompute_grad
     48 @@RevBlock
     49 @@rev_block
     50 @@safe_embedding_lookup_sparse
     51 @@scale_gradient
     52 @@separable_conv2d
     53 @@separable_convolution2d
     54 @@sequence_to_images
     55 @@softmax
     56 @@spatial_softmax
     57 @@stack
     58 @@unit_norm
     59 @@bow_encoder
     60 @@embed_sequence
     61 @@maxout
     62 
     63 @@apply_regularization
     64 @@l1_l2_regularizer
     65 @@l1_regularizer
     66 @@l2_regularizer
     67 @@sum_regularizer
     68 
     69 @@xavier_initializer
     70 @@xavier_initializer_conv2d
     71 @@variance_scaling_initializer
     72 
     73 @@optimize_loss
     74 
     75 @@summarize_activation
     76 @@summarize_tensor
     77 @@summarize_tensors
     78 @@summarize_collection
     79 
     80 @@summarize_activations
     81 
     82 @@bucketized_column
     83 @@check_feature_columns
     84 @@create_feature_spec_for_parsing
     85 @@crossed_column
     86 @@embedding_column
     87 @@scattered_embedding_column
     88 @@input_from_feature_columns
     89 @@transform_features
     90 @@joint_weighted_sum_from_feature_columns
     91 @@make_place_holder_tensors_for_base_features
     92 @@multi_class_target
     93 @@one_hot_column
     94 @@parse_feature_columns_from_examples
     95 @@parse_feature_columns_from_sequence_examples
     96 @@real_valued_column
     97 @@shared_embedding_columns
     98 @@sparse_column_with_hash_bucket
     99 @@sparse_column_with_integerized_feature
    100 @@sparse_column_with_keys
    101 @@sparse_column_with_vocabulary_file
    102 @@weighted_sparse_column
    103 @@weighted_sum_from_feature_columns
    104 @@infer_real_valued_columns
    105 @@sequence_input_from_feature_columns
    106 
    107 @@instance_norm
    108 """
    109 
    110 from __future__ import absolute_import
    111 from __future__ import division
    112 from __future__ import print_function
    113 
    114 # pylint: disable=unused-import,wildcard-import
    115 from tensorflow.contrib.layers.python.layers import *
    116 # pylint: enable=unused-import,wildcard-import
    117 
    118 from tensorflow.python.util.all_util import remove_undocumented
    119 
    120 _allowed_symbols = ['bias_add',
    121                     'conv2d',
    122                     'conv3d',
    123                     'elu',
    124                     'feature_column',
    125                     'instance_norm',
    126                     'legacy_fully_connected',
    127                     'legacy_linear',
    128                     'legacy_relu',
    129                     'OPTIMIZER_CLS_NAMES',
    130                     'OPTIMIZER_SUMMARIES',
    131                     'regression_target',
    132                     'SPARSE_FEATURE_CROSS_DEFAULT_HASH_KEY',
    133                     'summaries']
    134 
    135 remove_undocumented(__name__, _allowed_symbols)
    136