/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | group_by_reducer_dataset_op.cc | 156 b->BuildAttrValue(captured_key_func_->func(), &key_func); in AsGraphDefInternal() 158 b->BuildAttrValue(captured_init_func_->func(), &init_func); in AsGraphDefInternal() 160 b->BuildAttrValue(captured_reduce_func_->func(), &reduce_func); in AsGraphDefInternal() 162 b->BuildAttrValue(captured_finalize_func_->func(), &finalize_func); in AsGraphDefInternal() 165 b->BuildAttrValue(key_func_other_arguments_types, in AsGraphDefInternal() 168 b->BuildAttrValue(init_func_other_arguments_types, in AsGraphDefInternal() 171 b->BuildAttrValue(reduce_func_other_arguments_types, in AsGraphDefInternal() 174 b->BuildAttrValue(finalize_func_other_arguments_types, in AsGraphDefInternal()
|
D | scan_dataset_op.cc | 146 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 148 b->BuildAttrValue(state_types_, &state_types); in AsGraphDefInternal() 150 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 152 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal() 154 b->BuildAttrValue(use_default_device_, &use_default_device_attr); in AsGraphDefInternal()
|
D | snapshot_dataset_op.cc | 380 b->BuildAttrValue(compression_, &compression_attr); in AsGraphDefInternal() 383 b->BuildAttrValue(reader_prefix_, &reader_prefix_attr); in AsGraphDefInternal() 386 b->BuildAttrValue(writer_prefix_, &writer_prefix_attr); in AsGraphDefInternal() 389 b->BuildAttrValue(true, &hash_valid_attr); in AsGraphDefInternal() 392 b->BuildAttrValue(static_cast<int64>(hash_), &hash_attr); in AsGraphDefInternal() 395 b->BuildAttrValue(reader_func_->func(), &reader_func_attr); in AsGraphDefInternal() 398 b->BuildAttrValue(shard_func_->func(), &shard_func_attr); in AsGraphDefInternal() 401 b->BuildAttrValue(reader_func_other_args_types, in AsGraphDefInternal() 405 b->BuildAttrValue(shard_func_other_args_types, in AsGraphDefInternal() 1132 b->BuildAttrValue(compression_, &compression_attr); in AsGraphDefInternal() [all …]
|
D | group_by_window_dataset_op.cc | 157 b->BuildAttrValue(captured_key_func_->func(), &key_func); in AsGraphDefInternal() 159 b->BuildAttrValue(captured_reduce_func_->func(), &reduce_func); in AsGraphDefInternal() 161 b->BuildAttrValue(captured_window_size_func_->func(), &window_size_func); in AsGraphDefInternal() 164 b->BuildAttrValue(key_func_other_arguments_types, in AsGraphDefInternal() 167 b->BuildAttrValue(reduce_func_other_arguments_types, in AsGraphDefInternal() 170 b->BuildAttrValue(window_size_func_other_arguments_types, in AsGraphDefInternal()
|
D | parse_example_dataset_op.cc | 307 b->BuildAttrValue(sparse_keys_, &sparse_keys_attr); in AsGraphDefInternal() 311 b->BuildAttrValue(dense_keys_, &dense_keys_attr); in AsGraphDefInternal() 315 b->BuildAttrValue(sparse_types_, &sparse_types_attr); in AsGraphDefInternal() 319 b->BuildAttrValue(dense_types_, &dense_attr); in AsGraphDefInternal() 323 b->BuildAttrValue(dense_shapes_, &dense_shapes_attr); in AsGraphDefInternal() 328 b->BuildAttrValue(deterministic_.IsNondeterministic(), &sloppy_attr); in AsGraphDefInternal() 333 b->BuildAttrValue(deterministic_.String(), &deterministic_attr); in AsGraphDefInternal() 339 b->BuildAttrValue(ragged_keys_, &ragged_keys_attr); in AsGraphDefInternal() 343 b->BuildAttrValue(ragged_value_types_, &ragged_value_types_attr); in AsGraphDefInternal() 347 b->BuildAttrValue(ragged_split_types_, &ragged_split_types_attr); in AsGraphDefInternal()
|
D | take_while_dataset_op.cc | 108 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal() 111 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | choose_fastest_branch_dataset_op.cc | 296 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 300 b->BuildAttrValue(num_elements_per_branch_, in AsGraphDefInternal() 310 b->BuildAttrValue(funcs, &branches_attr); in AsGraphDefInternal() 314 b->BuildAttrValue(other_arguments_lengths, &other_arguments_lengths_attr); in AsGraphDefInternal()
|
D | io_ops.cc | 229 b->BuildAttrValue(compression_, &compression_attr); in AsGraphDefInternal() 233 b->BuildAttrValue(captured_func_->func(), &reader_func_attr); in AsGraphDefInternal() 236 b->BuildAttrValue(reader_func_other_args_types, in AsGraphDefInternal()
|
D | ignore_errors_dataset_op.cc | 86 b->BuildAttrValue<bool>(log_warning_, &log_warning_attr); in AsGraphDefInternal()
|
D | map_and_batch_dataset_op.cc | 165 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 167 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 169 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
|
D | parallel_interleave_dataset_op.cc | 206 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 211 b->BuildAttrValue(deterministic_.String(), &deterministic_attr); in AsGraphDefInternal() 216 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | map_dataset_op.cc | 107 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal() 111 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 115 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal() 120 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
|
D | model_dataset_op.cc | 103 b->BuildAttrValue(static_cast<int64>(algorithm_), &algorithm_attr); in AsGraphDefInternal() 105 b->BuildAttrValue(cpu_budget_, &cpu_budget_attr); in AsGraphDefInternal() 107 b->BuildAttrValue(ram_budget_, &ram_budget_attr); in AsGraphDefInternal()
|
D | parallel_map_dataset_op.cc | 160 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal() 165 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 170 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal() 177 b->BuildAttrValue(deterministic_.IsNondeterministic(), &sloppy_attr); in AsGraphDefInternal() 182 b->BuildAttrValue(deterministic_.String(), &deterministic_attr); in AsGraphDefInternal() 188 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
|
D | filter_dataset_op.cc | 99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 101 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | flat_map_dataset_op.cc | 99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 101 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | padded_batch_dataset_op.cc | 164 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal() 167 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal() 170 b->BuildAttrValue<int64>(padded_shapes_.size(), &N); in AsGraphDefInternal()
|
D | prefetch_dataset_op.cc | 116 b->BuildAttrValue(slack_period_, &slack_period_attr); in AsGraphDefInternal() 118 b->BuildAttrValue(legacy_autotune_, &legacy_autotune_attr); in AsGraphDefInternal() 120 b->BuildAttrValue(buffer_size_min_, &buffer_size_min_attr); in AsGraphDefInternal()
|
D | interleave_dataset_op.cc | 115 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 117 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | tensor_dataset_op.cc | 91 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
|
D | shuffle_dataset_op.cc | 477 b->BuildAttrValue(seed_generator_->reshuffle_each_iteration(), in AsGraphDefInternal() 595 b->BuildAttrValue(seed_generator_->reshuffle_each_iteration(), in AsGraphDefInternal() 785 b->BuildAttrValue(seed_generator_->reshuffle_each_iteration(), in AsGraphDefInternal() 846 b->BuildAttrValue(seed_generator_->reshuffle_each_iteration(), in AsGraphDefInternal()
|
D | tensor_slice_dataset_op.cc | 104 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
|
D | shard_dataset_op.cc | 108 b->BuildAttrValue(require_non_empty_, &require_non_empty_attr); in AsGraphDefInternal()
|
D | batch_dataset_op.cc | 140 b->BuildAttrValue(parallel_copy_, ¶llel_copy); in AsGraphDefInternal()
|
D | parallel_interleave_dataset_op.cc | 285 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal() 289 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal() 294 b->BuildAttrValue(deterministic_.IsNondeterministic(), &sloppy_attr); in AsGraphDefInternal() 299 b->BuildAttrValue(deterministic_.String(), &deterministic_attr); in AsGraphDefInternal()
|