Searched refs:MapDatasetParams (Results 1 – 3 of 3) sorted by relevance
27 MapDatasetParams MapDatasetParams1() { in MapDatasetParams1()28 auto map_dataset_params_0 = MapDatasetParams( in MapDatasetParams1()40 return MapDatasetParams( in MapDatasetParams1()54 MapDatasetParams MapDatasetParams2() { in MapDatasetParams2()63 return MapDatasetParams( in MapDatasetParams2()79 MapDatasetParams MapDatasetParams3() { in MapDatasetParams3()80 return MapDatasetParams( in MapDatasetParams3()94 std::vector<GetNextTestCase<MapDatasetParams>> GetNextTestCases() { in GetNextTestCases()106 ITERATOR_GET_NEXT_TEST_P(MapDatasetOpTest, MapDatasetParams, GetNextTestCases()) in ITERATOR_GET_NEXT_TEST_P() argument133 std::vector<CardinalityTestCase<MapDatasetParams>> CardinalityTestCases() { in CardinalityTestCases()[all …]
1035 std::vector<Tensor> MapDatasetParams::GetInputTensors() const { in GetInputTensors()1039 Status MapDatasetParams::GetInputNames(std::vector<string>* input_names) const { in GetInputNames()1048 Status MapDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()1059 string MapDatasetParams::dataset_type() const { in dataset_type()1063 std::vector<FunctionDef> MapDatasetParams::func_lib() const { in func_lib()
241 class MapDatasetParams : public DatasetParams {244 MapDatasetParams(T input_dataset_params, std::vector<Tensor> other_arguments, in MapDatasetParams() function