tensorflow.data.experimental.pbtxt 7.2 KB
Newer Older
1 2
path: "tensorflow.data.experimental"
tf_module {
3 4 5 6
  member {
    name: "AUTOTUNE"
    mtype: "<type \'int\'>"
  }
7 8 9 10
  member {
    name: "AutoShardPolicy"
    mtype: "<class \'enum.EnumMeta\'>"
  }
11 12 13 14 15 16
  member {
    name: "CheckpointInputPipelineHook"
    mtype: "<type \'type\'>"
  }
  member {
    name: "CsvDataset"
G
Geoffrey Irving 已提交
17
    mtype: "<type \'type\'>"
18
  }
19
  member {
20 21 22 23
    name: "DistributeOptions"
    mtype: "<type \'type\'>"
  }
  member {
24 25 26
    name: "INFINITE_CARDINALITY"
    mtype: "<type \'int\'>"
  }
27 28 29 30
  member {
    name: "MapVectorizationOptions"
    mtype: "<type \'type\'>"
  }
31 32 33 34
  member {
    name: "OptimizationOptions"
    mtype: "<type \'type\'>"
  }
35 36 37 38 39 40
  member {
    name: "Optional"
    mtype: "<type \'type\'>"
  }
  member {
    name: "RandomDataset"
G
Geoffrey Irving 已提交
41
    mtype: "<type \'type\'>"
42 43 44 45 46 47 48
  }
  member {
    name: "Reducer"
    mtype: "<type \'type\'>"
  }
  member {
    name: "SqlDataset"
G
Geoffrey Irving 已提交
49
    mtype: "<type \'type\'>"
50 51 52 53 54
  }
  member {
    name: "StatsAggregator"
    mtype: "<type \'type\'>"
  }
55 56 57 58
  member {
    name: "StatsOptions"
    mtype: "<type \'type\'>"
  }
59 60 61 62
  member {
    name: "TFRecordWriter"
    mtype: "<type \'type\'>"
  }
J
Jiri Simsa 已提交
63 64 65 66
  member {
    name: "ThreadingOptions"
    mtype: "<type \'type\'>"
  }
67 68 69 70
  member {
    name: "UNKNOWN_CARDINALITY"
    mtype: "<type \'int\'>"
  }
71 72 73 74 75 76
  member_method {
    name: "Counter"
    argspec: "args=[\'start\', \'step\', \'dtype\'], varargs=None, keywords=None, defaults=[\'0\', \'1\', \"<dtype: \'int64\'>\"], "
  }
  member_method {
    name: "bucket_by_sequence_length"
77
    argspec: "args=[\'element_length_func\', \'bucket_boundaries\', \'bucket_batch_sizes\', \'padded_shapes\', \'padding_values\', \'pad_to_bucket_boundary\', \'no_padding\', \'drop_remainder\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'False\', \'False\'], "
78
  }
79 80 81 82
  member_method {
    name: "bytes_produced_stats"
    argspec: "args=[\'tag\'], varargs=None, keywords=None, defaults=None"
  }
83 84 85 86
  member_method {
    name: "cardinality"
    argspec: "args=[\'dataset\'], varargs=None, keywords=None, defaults=None"
  }
87 88 89 90 91 92 93 94
  member_method {
    name: "choose_from_datasets"
    argspec: "args=[\'datasets\', \'choice_dataset\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "copy_to_device"
    argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/cpu:0\'], "
  }
95 96 97 98
  member_method {
    name: "dense_to_ragged_batch"
    argspec: "args=[\'batch_size\', \'drop_remainder\', \'row_splits_dtype\'], varargs=None, keywords=None, defaults=[\'False\', \"<dtype: \'int64\'>\"], "
  }
99 100 101 102 103 104 105 106
  member_method {
    name: "dense_to_sparse_batch"
    argspec: "args=[\'batch_size\', \'row_shape\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "enumerate_dataset"
    argspec: "args=[\'start\'], varargs=None, keywords=None, defaults=[\'0\'], "
  }
107 108 109 110
  member_method {
    name: "from_variant"
    argspec: "args=[\'variant\', \'structure\'], varargs=None, keywords=None, defaults=None"
  }
111 112 113 114 115 116 117 118
  member_method {
    name: "get_next_as_optional"
    argspec: "args=[\'iterator\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "get_single_element"
    argspec: "args=[\'dataset\'], varargs=None, keywords=None, defaults=None"
  }
119 120 121 122
  member_method {
    name: "get_structure"
    argspec: "args=[\'dataset_or_iterator\'], varargs=None, keywords=None, defaults=None"
  }
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140
  member_method {
    name: "group_by_reducer"
    argspec: "args=[\'key_func\', \'reducer\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "group_by_window"
    argspec: "args=[\'key_func\', \'reduce_func\', \'window_size\', \'window_size_func\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
  }
  member_method {
    name: "ignore_errors"
    argspec: "args=[], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "latency_stats"
    argspec: "args=[\'tag\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "make_batched_features_dataset"
141
    argspec: "args=[\'file_pattern\', \'batch_size\', \'features\', \'reader\', \'label_key\', \'reader_args\', \'num_epochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'reader_num_threads\', \'parser_num_threads\', \'sloppy_ordering\', \'drop_final_batch\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'True\', \'10000\', \'None\', \'None\', \'None\', \'None\', \'False\', \'False\'], "
142 143 144
  }
  member_method {
    name: "make_csv_dataset"
145
    argspec: "args=[\'file_pattern\', \'batch_size\', \'column_names\', \'column_defaults\', \'label_name\', \'select_columns\', \'field_delim\', \'use_quote_delim\', \'na_value\', \'header\', \'num_epochs\', \'shuffle\', \'shuffle_buffer_size\', \'shuffle_seed\', \'prefetch_buffer_size\', \'num_parallel_reads\', \'sloppy\', \'num_rows_for_inference\', \'compression_type\', \'ignore_errors\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \',\', \'True\', \'\', \'True\', \'None\', \'True\', \'10000\', \'None\', \'None\', \'None\', \'False\', \'100\', \'None\', \'False\'], "
146 147 148
  }
  member_method {
    name: "make_saveable_from_iterator"
149
    argspec: "args=[\'iterator\', \'external_state_policy\'], varargs=None, keywords=None, defaults=[\'fail\'], "
150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182
  }
  member_method {
    name: "map_and_batch"
    argspec: "args=[\'map_func\', \'batch_size\', \'num_parallel_batches\', \'drop_remainder\', \'num_parallel_calls\'], varargs=None, keywords=None, defaults=[\'None\', \'False\', \'None\'], "
  }
  member_method {
    name: "parallel_interleave"
    argspec: "args=[\'map_func\', \'cycle_length\', \'block_length\', \'sloppy\', \'buffer_output_elements\', \'prefetch_input_elements\'], varargs=None, keywords=None, defaults=[\'1\', \'False\', \'None\', \'None\'], "
  }
  member_method {
    name: "parse_example_dataset"
    argspec: "args=[\'features\', \'num_parallel_calls\'], varargs=None, keywords=None, defaults=[\'1\'], "
  }
  member_method {
    name: "prefetch_to_device"
    argspec: "args=[\'device\', \'buffer_size\'], varargs=None, keywords=None, defaults=[\'None\'], "
  }
  member_method {
    name: "rejection_resample"
    argspec: "args=[\'class_func\', \'target_dist\', \'initial_dist\', \'seed\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
  }
  member_method {
    name: "sample_from_datasets"
    argspec: "args=[\'datasets\', \'weights\', \'seed\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
  }
  member_method {
    name: "scan"
    argspec: "args=[\'initial_state\', \'scan_func\'], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "shuffle_and_repeat"
    argspec: "args=[\'buffer_size\', \'count\', \'seed\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
  }
183 184 185 186
  member_method {
    name: "take_while"
    argspec: "args=[\'predicate\'], varargs=None, keywords=None, defaults=None"
  }
187 188 189 190
  member_method {
    name: "to_variant"
    argspec: "args=[\'dataset\'], varargs=None, keywords=None, defaults=None"
  }
191 192 193 194 195 196 197 198 199
  member_method {
    name: "unbatch"
    argspec: "args=[], varargs=None, keywords=None, defaults=None"
  }
  member_method {
    name: "unique"
    argspec: "args=[], varargs=None, keywords=None, defaults=None"
  }
}