Home
last modified time | relevance | path

Searched refs:MapDatasetOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op.cc31 /* static */ constexpr const char* const MapDatasetOp::kDatasetType;
32 /* static */ constexpr const char* const MapDatasetOp::kInputDataset;
33 /* static */ constexpr const char* const MapDatasetOp::kOtherArguments;
34 /* static */ constexpr const char* const MapDatasetOp::kFunc;
35 /* static */ constexpr const char* const MapDatasetOp::kTarguments;
36 /* static */ constexpr const char* const MapDatasetOp::kOutputTypes;
37 /* static */ constexpr const char* const MapDatasetOp::kOutputShapes;
38 /* static */ constexpr const char* const MapDatasetOp::kUseInterOpParallelism;
39 /* static */ constexpr const char* const MapDatasetOp::kPreserveCardinality;
41 class MapDatasetOp::Dataset : public DatasetBase {
[all …]
Dmap_dataset_op.h24 class MapDatasetOp : public UnaryDatasetOpKernel {
38 explicit MapDatasetOp(OpKernelConstruction* ctx);
Dmap_dataset_op_test.cc118 CheckDatasetTypeString(name_utils::OpName(MapDatasetOp::kDatasetType))); in TEST_F()
161 MapDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in TEST_F()
Ddataset_test_base.cc1040 input_names->emplace_back(MapDatasetOp::kInputDataset); in GetInputNames()
1043 absl::StrCat(MapDatasetOp::kOtherArguments, "_", i)); in GetInputNames()
1050 {MapDatasetOp::kFunc, func_}, in GetAttributes()
1051 {MapDatasetOp::kTarguments, type_arguments_}, in GetAttributes()
1052 {MapDatasetOp::kOutputShapes, output_shapes_}, in GetAttributes()
1053 {MapDatasetOp::kOutputTypes, output_dtypes_}, in GetAttributes()
1054 {MapDatasetOp::kUseInterOpParallelism, use_inter_op_parallelism_}, in GetAttributes()
1055 {MapDatasetOp::kPreserveCardinality, preserve_cardinality_}}; in GetAttributes()
1060 return MapDatasetOp::kDatasetType; in dataset_type()