/external/tensorflow/tensorflow/core/kernels/data/ |
D | generator_dataset_op.cc | 49 Dataset(OpKernelContext* ctx, std::unique_ptr<CapturedFunction> init_func, in Dataset() 50 std::unique_ptr<CapturedFunction> next_func, in Dataset() 51 std::unique_ptr<CapturedFunction> finalize_func, in Dataset() 187 const std::unique_ptr<CapturedFunction> init_func_; 188 const std::unique_ptr<CapturedFunction> next_func_; 189 const std::unique_ptr<CapturedFunction> finalize_func_; 209 std::unique_ptr<CapturedFunction> init_func; in MakeDataset() 210 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, init_func_metadata_, in MakeDataset() 213 std::unique_ptr<CapturedFunction> next_func; in MakeDataset() 214 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, next_func_metadata_, in MakeDataset() [all …]
|
D | captured_function.h | 40 class CapturedFunction; variable 129 class CapturedFunction { 136 std::unique_ptr<CapturedFunction>* out_function); 143 std::unique_ptr<CapturedFunction>* out_function); 189 CapturedFunction(std::shared_ptr<const FunctionMetadata> metadata, 197 TF_DISALLOW_COPY_AND_ASSIGN(CapturedFunction); 218 CapturedFunction* captured_func, bool is_multi_device, 277 CapturedFunction* captured_func, bool is_multi_device); 289 CapturedFunction* const captured_func_; // Not owned.
|
D | captured_function.cc | 113 Status GetCapturedInput(const CapturedFunction* const func, int index, in GetCapturedInput() 127 const CapturedFunction* const func, in RunShortCircuit() 146 const CapturedFunction* const func, in RunShortCircuit() 544 Status CapturedFunction::Create( in Create() 547 std::unique_ptr<CapturedFunction>* out_function) { in Create() 556 Status CapturedFunction::Create( in Create() 559 std::unique_ptr<CapturedFunction>* out_function) { in Create() 561 new CapturedFunction(std::move(metadata), std::move(captured_inputs))); in Create() 565 Status CapturedFunction::AddToGraph( in AddToGraph() 588 Status CapturedFunction::Instantiate( in Instantiate() [all …]
|
D | map_dataset_op.cc | 44 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 209 const std::unique_ptr<CapturedFunction> captured_func_; 229 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 231 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | filter_dataset_op.cc | 51 std::unique_ptr<CapturedFunction> captured_func) in Dataset() 242 const std::unique_ptr<CapturedFunction> captured_func_; 256 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 258 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | flat_map_dataset_op.cc | 49 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 270 const std::unique_ptr<CapturedFunction> captured_func_; 285 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 287 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | interleave_dataset_op.cc | 54 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() 304 const std::unique_ptr<CapturedFunction> captured_func_; 337 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 339 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | parallel_map_dataset_op.cc | 78 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 671 const std::unique_ptr<CapturedFunction> captured_func_; 721 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 723 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | parallel_interleave_dataset_op.cc | 162 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() 1557 const std::unique_ptr<CapturedFunction> captured_func_; 1655 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 1657 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | iterator_ops.cc | 702 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() 703 TF_RETURN_IF_ERROR(CapturedFunction::Create( in DoCompute()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | group_by_reducer_dataset_op.cc | 52 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset() 53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, key_func_metadata_, in MakeDataset() 56 std::unique_ptr<CapturedFunction> captured_init_func; in MakeDataset() 57 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, init_func_metadata_, in MakeDataset() 60 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset() 61 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, reduce_func_metadata_, in MakeDataset() 64 std::unique_ptr<CapturedFunction> captured_finalize_func; in MakeDataset() 66 CapturedFunction::Create(ctx, finalize_func_metadata_, in MakeDataset() 80 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset() 81 std::unique_ptr<CapturedFunction> captured_init_func, in Dataset() [all …]
|
D | group_by_window_dataset_op.cc | 50 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset() 51 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, key_func_metadata_, in MakeDataset() 55 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset() 56 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, reduce_func_metadata_, in MakeDataset() 60 std::unique_ptr<CapturedFunction> captured_window_size_func; in MakeDataset() 62 CapturedFunction::Create(ctx, window_size_func_metadata_, in MakeDataset() 76 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset() 77 std::unique_ptr<CapturedFunction> captured_reduce_func, in Dataset() 78 std::unique_ptr<CapturedFunction> captured_window_size_func, in Dataset() 530 const std::unique_ptr<CapturedFunction> captured_key_func_; [all …]
|
D | io_ops.cc | 59 std::unique_ptr<CapturedFunction> captured_func; in DoCompute() 60 TF_RETURN_IF_ERROR(CapturedFunction::Create( in DoCompute() 73 std::unique_ptr<CapturedFunction> captured_func, in WriteData() 182 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 339 const std::unique_ptr<CapturedFunction> captured_func_; 359 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 361 ctx, CapturedFunction::Create(ctx, func_metadata_, kReaderFuncOtherArgs, in MakeDataset()
|
D | take_while_dataset_op.cc | 45 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 47 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset() 56 std::unique_ptr<CapturedFunction> captured_func) in Dataset() 205 const std::unique_ptr<CapturedFunction> captured_func_;
|
D | scan_dataset_op.cc | 59 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 61 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset() 75 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 306 const std::unique_ptr<CapturedFunction> captured_func_;
|
D | choose_fastest_branch_dataset_op.cc | 178 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs( in MakeDataset() 192 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, func_metadatas_[i], in MakeDataset() 205 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs, in Dataset() 568 const std::vector<std::unique_ptr<CapturedFunction>> captured_funcs_;
|
D | io_ops.h | 53 std::unique_ptr<CapturedFunction> captured_func,
|
D | map_and_batch_dataset_op.cc | 88 std::unique_ptr<CapturedFunction> captured_func, in Dataset() 686 const std::unique_ptr<CapturedFunction> captured_func_; 719 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 721 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | snapshot_dataset_op.cc | 123 std::unique_ptr<CapturedFunction> reader_func, 124 std::unique_ptr<CapturedFunction> shard_func); 156 std::unique_ptr<CapturedFunction> reader_func_; 157 std::unique_ptr<CapturedFunction> shard_func_; 312 std::unique_ptr<CapturedFunction> reader_func, in Dataset() 313 std::unique_ptr<CapturedFunction> shard_func) in Dataset() 887 std::unique_ptr<CapturedFunction> reader_func; in MakeDataset() 889 CapturedFunction::Create(ctx, reader_func_metadata_, in MakeDataset() 891 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset() 893 CapturedFunction::Create(ctx, shard_func_metadata_, in MakeDataset()
|
D | parallel_interleave_dataset_op.cc | 96 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset() 1160 const std::unique_ptr<CapturedFunction> captured_func_; 1226 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset() 1228 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 910 friend class CapturedFunction; variable
|