/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | group_by_reducer_dataset_op.cc | 144 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()
|
D | parse_example_dataset_op.cc | 236 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()
|
D | scan_dataset_op.cc | 133 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()
|
D | group_by_window_dataset_op.cc | 137 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()
|
D | choose_fastest_branch_dataset_op.cc | 280 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()
|
D | take_while_dataset_op.cc | 154 b->BuildAttrValue(func_, &f_attr); in AsGraphDefInternal() 157 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | map_and_batch_dataset_op.cc | 231 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()
|
D | choose_fastest_dataset_op.cc | 172 b->BuildAttrValue(num_experiments_, &num_experiments_attr); in AsGraphDefInternal()
|
D | parallel_interleave_dataset_op.cc | 167 b->BuildAttrValue(interleave_func_, &f); in AsGraphDefInternal() 169 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | numa_map_and_batch_dataset_op.cc | 182 b->BuildAttrValue(func_, &f); in AsGraphDefInternal() 184 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | parallel_map_dataset_op.cc | 170 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()
|
D | map_dataset_op.cc | 163 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()
|
D | flat_map_dataset_op.cc | 109 b->BuildAttrValue(func_, &f); in AsGraphDefInternal() 111 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | filter_dataset_op.cc | 151 b->BuildAttrValue(func_, &f); in AsGraphDefInternal() 153 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | interleave_dataset_op.cc | 135 b->BuildAttrValue(func_, &f); in AsGraphDefInternal() 137 b->BuildAttrValue(other_arguments_types, &other_arguments_types_attr); in AsGraphDefInternal()
|
D | padded_batch_dataset_op.cc | 197 b->BuildAttrValue(output_dtypes(), &output_types); in AsGraphDefInternal() 200 b->BuildAttrValue<int64>(padded_shapes_.size(), &N); in AsGraphDefInternal()
|
D | tensor_dataset_op.cc | 90 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
|
D | tensor_slice_dataset_op.cc | 115 b->BuildAttrValue(dtypes_, &dtypes); in AsGraphDefInternal()
|
D | parallel_interleave_dataset_op.cc | 179 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()
|
D | shuffle_dataset_op.cc | 552 b->BuildAttrValue(true, &reshuffle_each_iteration); in AsGraphDefInternal() 604 b->BuildAttrValue(false, &reshuffle_each_iteration); in AsGraphDefInternal()
|
D | sparse_tensor_slice_dataset_op.cc | 74 b->BuildAttrValue(sparse_tensor_.dtype(), &val_dtype); in AsGraphDefInternal()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 192 void BuildAttrValue(const T& value, AttrValue* attr) { in BuildAttrValue() function
|