Home
last modified time | relevance | path

Searched refs:func_metadata_ (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dtake_while_dataset_op.cc37 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
Dio_ops.h62 std::shared_ptr<FunctionMetadata> func_metadata_; variable
89 std::shared_ptr<FunctionMetadata> func_metadata_; variable
Dio_ops.cc40 &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()
Dmap_and_batch_dataset_op.h51 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dscan_dataset_op.cc43 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
Dparallel_interleave_dataset_op.h57 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dmap_and_batch_dataset_op.cc694 &func_metadata_)); in MapAndBatchDatasetOp()
721 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dparallel_interleave_dataset_op.cc1177 &func_metadata_)); in ParallelInterleaveDatasetOp()
1228 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
/external/tensorflow/tensorflow/core/kernels/data/
Dfilter_dataset_op.cc248 &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()
Dfilter_dataset_op.h42 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dflat_map_dataset_op.h45 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dmap_dataset_op.h46 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dinterleave_dataset_op.h47 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dparallel_map_dataset_op.h51 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dparallel_interleave_dataset_op.h53 std::shared_ptr<FunctionMetadata> func_metadata_ = nullptr; variable
Dmap_dataset_op.cc220 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in MapDatasetOp()
231 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dflat_map_dataset_op.cc278 &func_metadata_)); in FlatMapDatasetOp()
287 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dinterleave_dataset_op.cc315 &func_metadata_)); in InterleaveDatasetOp()
339 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dparallel_map_dataset_op.cc681 FunctionMetadata::Create(ctx, kFunc, params, &func_metadata_)); in ParallelMapDatasetOp()
723 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Diterator_ops.cc687 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
Dparallel_interleave_dataset_op.cc1575 &func_metadata_)); in ParallelInterleaveDatasetOp()
1657 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()