Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc144 b->BuildAttrValue(this->key_func(), &key_func); in AsGraphDefInternal()
146 b->BuildAttrValue(this->init_func(), &init_func); in AsGraphDefInternal()
148 b->BuildAttrValue(this->reduce_func(), &reduce_func); in AsGraphDefInternal()
150 b->BuildAttrValue(this->finalize_func(), &finalize_func); in AsGraphDefInternal()
153 b->BuildAttrValue(key_func_other_arguments_types, in AsGraphDefInternal()
156 b->BuildAttrValue(init_func_other_arguments_types, in AsGraphDefInternal()
159 b->BuildAttrValue(reduce_func_other_arguments_types, in AsGraphDefInternal()
162 b->BuildAttrValue(finalize_func_other_arguments_types, in AsGraphDefInternal()
Dparse_example_dataset_op.cc236 b->BuildAttrValue(sparse_keys_, &sparse_keys_attr); in AsGraphDefInternal()
237 b->BuildAttrValue(dense_keys_, &dense_keys_attr); in AsGraphDefInternal()
238 b->BuildAttrValue(sparse_types_, &sparse_types_attr); in AsGraphDefInternal()
239 b->BuildAttrValue(dense_types_, &dense_attr); in AsGraphDefInternal()
240 b->BuildAttrValue(dense_shapes_, &dense_shapes_attr); in AsGraphDefInternal()
241 b->BuildAttrValue(sloppy_, &sloppy_attr); in AsGraphDefInternal()
Dscan_dataset_op.cc133 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
135 b->BuildAttrValue(state_types_, &state_types); in AsGraphDefInternal()
137 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
139 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
Dgroup_by_window_dataset_op.cc137 b->BuildAttrValue(key_func_, &key_func); in AsGraphDefInternal()
139 b->BuildAttrValue(reduce_func_, &reduce_func); in AsGraphDefInternal()
141 b->BuildAttrValue(window_size_func_, &window_size_func); in AsGraphDefInternal()
144 b->BuildAttrValue(key_func_other_arguments_types, in AsGraphDefInternal()
147 b->BuildAttrValue(reduce_func_other_arguments_types, in AsGraphDefInternal()
150 b->BuildAttrValue(window_size_func_other_arguments_types, in AsGraphDefInternal()
Dchoose_fastest_branch_dataset_op.cc280 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
284 b->BuildAttrValue(num_elements_per_branch_, in AsGraphDefInternal()
289 b->BuildAttrValue(funcs_, &branches_attr); in AsGraphDefInternal()
296 b->BuildAttrValue(other_arguments_lengths, &other_arguments_lengths_attr); in AsGraphDefInternal()
Dtake_while_dataset_op.cc154 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
157 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc231 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
233 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
235 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
Dchoose_fastest_dataset_op.cc172 b->BuildAttrValue(num_experiments_, &num_experiments_attr); in AsGraphDefInternal()
Dparallel_interleave_dataset_op.cc167 b->BuildAttrValue(interleave_func_, &f); in AsGraphDefInternal()
169 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dnuma_map_and_batch_dataset_op.cc182 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
184 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc170 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
174 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
178 b->BuildAttrValue(use_inter_op_parallelism_, in AsGraphDefInternal()
183 b->BuildAttrValue(sloppy_, &sloppy_attr); in AsGraphDefInternal()
187 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
Dmap_dataset_op.cc163 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal()
167 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
171 b->BuildAttrValue(use_inter_op_parallelism_, in AsGraphDefInternal()
176 b->BuildAttrValue(preserve_cardinality_, &preserve_cardinality_attr); in AsGraphDefInternal()
Dflat_map_dataset_op.cc109 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
111 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dfilter_dataset_op.cc151 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
153 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dinterleave_dataset_op.cc135 b->BuildAttrValue(func_, &f); in AsGraphDefInternal()
137 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
Dpadded_batch_dataset_op.cc197 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal()
200 b->BuildAttrValue<int64>(padded_shapes_.size(), &N); in AsGraphDefInternal()
Dtensor_dataset_op.cc90 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
Dtensor_slice_dataset_op.cc115 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
Dparallel_interleave_dataset_op.cc179 b->BuildAttrValue(interleave_func_, &f); in AsGraphDefInternal()
181 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
183 b->BuildAttrValue(sloppy_, &sloppy_attr); in AsGraphDefInternal()
Dshuffle_dataset_op.cc552 b->BuildAttrValue(true, &reshuffle_each_iteration); in AsGraphDefInternal()
604 b->BuildAttrValue(false, &reshuffle_each_iteration); in AsGraphDefInternal()
Dsparse_tensor_slice_dataset_op.cc74 b->BuildAttrValue(sparse_tensor_.dtype(), &val_dtype); in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h192 void BuildAttrValue(const T& value, AttrValue* attr) { in BuildAttrValue() function