/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | take_while_dataset_op.cc | 37 ctx, "predicate", /*params=*/{}, &func_metadata_)); in TakeWhileDatasetOp() 38 OP_REQUIRES(ctx, func_metadata_->short_circuit_info().indices.size() <= 1, in TakeWhileDatasetOp() 47 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset() 208 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; member in tensorflow::data::experimental::__anonaa7f23910111::TakeWhileDatasetOp
|
D | io_ops.h | 62 std::shared_ptr<FunctionMetadata> func_metadata_; variable 89 std::shared_ptr<FunctionMetadata> func_metadata_; variable
|
D | io_ops.cc | 40 &func_metadata_)); in SaveDatasetOp() 61 ctx, func_metadata_, kShardFuncOtherArgs, &captured_func)); in DoCompute() 352 &func_metadata_)); in LoadDatasetOp() 361 ctx, CapturedFunction::Create(ctx, func_metadata_, kReaderFuncOtherArgs, in MakeDataset()
|
D | map_and_batch_dataset_op.h | 51 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | scan_dataset_op.cc | 43 FunctionMetadata::Create(ctx, "f", params, &func_metadata_)); in ScanDatasetOp() 61 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset() 314 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp
|
D | parallel_interleave_dataset_op.h | 57 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | map_and_batch_dataset_op.cc | 694 &func_metadata_)); in MapAndBatchDatasetOp() 721 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | parallel_interleave_dataset_op.cc | 1177 &func_metadata_)); in ParallelInterleaveDatasetOp() 1228 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | filter_dataset_op.cc | 248 &func_metadata_)); in FilterDatasetOp() 249 OP_REQUIRES(ctx, func_metadata_->short_circuit_info().indices.size() <= 1, in FilterDatasetOp() 258 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | filter_dataset_op.h | 42 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | flat_map_dataset_op.h | 45 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | map_dataset_op.h | 46 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | interleave_dataset_op.h | 47 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | parallel_map_dataset_op.h | 51 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | parallel_interleave_dataset_op.h | 53 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
|
D | map_dataset_op.cc | 220 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in MapDatasetOp() 231 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | flat_map_dataset_op.cc | 278 &func_metadata_)); in FlatMapDatasetOp() 287 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | interleave_dataset_op.cc | 315 &func_metadata_)); in InterleaveDatasetOp() 339 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | parallel_map_dataset_op.cc | 681 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in ParallelMapDatasetOp() 723 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|
D | iterator_ops.cc | 687 FunctionMetadata::Create(ctx, "f", params, &func_metadata_)); in ReduceDatasetOp() 704 ctx, func_metadata_, "other_arguments", &captured_func)); in DoCompute() 765 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; member in tensorflow::data::__anonadcc0d0f0711::ReduceDatasetOp
|
D | parallel_interleave_dataset_op.cc | 1575 &func_metadata_)); in ParallelInterleaveDatasetOp() 1657 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
|