Home
last modified time | relevance | path

Searched refs:BuildAttrValue (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc156 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()
Dscan_dataset_op.cc146 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()
Dsnapshot_dataset_op.cc380 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 …]
Dgroup_by_window_dataset_op.cc157 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()
Dparse_example_dataset_op.cc307 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()
Dtake_while_dataset_op.cc108 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
111 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dchoose_fastest_branch_dataset_op.cc296 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()
Dio_ops.cc229 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()
Dignore_errors_dataset_op.cc86 b->BuildAttrValue<bool>(log_warning_, &log_warning_attr); in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc165 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()
Dparallel_interleave_dataset_op.cc206 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/
Dmap_dataset_op.cc107 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()
Dmodel_dataset_op.cc103 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()
Dparallel_map_dataset_op.cc160 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()
Dfilter_dataset_op.cc99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
101 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dflat_map_dataset_op.cc99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
101 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dpadded_batch_dataset_op.cc164 b->BuildAttrValue(parallel_copy_, &parallel_copy); in AsGraphDefInternal()
167 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal()
170 b->BuildAttrValue<int64>(padded_shapes_.size(), &N); in AsGraphDefInternal()
Dprefetch_dataset_op.cc116 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()
Dinterleave_dataset_op.cc115 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
117 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dtensor_dataset_op.cc91 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
Dshuffle_dataset_op.cc477 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()
Dtensor_slice_dataset_op.cc104 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
Dshard_dataset_op.cc108 b->BuildAttrValue(require_non_empty_, &require_non_empty_attr); in AsGraphDefInternal()
Dbatch_dataset_op.cc140 b->BuildAttrValue(parallel_copy_, &parallel_copy); in AsGraphDefInternal()
Dparallel_interleave_dataset_op.cc285 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()

12