tensorflow/core/kernels/data/flat_map_dataset_op_test.cc
// Test case 2: test the case if the function does not return a single scalar
// of dtype DT_VARIANT.
Test case 2: test the case if the function does not return a single scalar
of dtype DT_VARIANT.
Test case 2 : test the case if the function does not return a single scalar of dtype DT_VARIANT .
FlatMapDatasetParams InvalidFlatMapDatasetParams() {
  auto tensor_slice_dataset_params = TensorSliceDatasetParams(
      {CreateTensor<int64_t>(TensorShape{3, 3, 1},
                                            {0, 1, 2, 3, 4, 5, 6, 7, 8})},
      "tensor_slice");
  auto func = FunctionDefHelper::FunctionRef( "NonZero",
                                              {{"T", DT_INT64}});
  return FlatMapDatasetParams(std::move(tensor_slice_dataset_params),
                              {},
                              func,
                              {test::function::NonZero()},
                              {},
                              {DT_INT64},
                              {PartialTensorShape({1})},
                              kNodeName);
}
FlatMapDatasetParams InvalidFlatMapDatasetParams ( ) { auto tensor_slice_dataset_params = TensorSliceDatasetParams ( { CreateTensor < int64_t > ( TensorShape { 3 , 3 , 1 } , { 0 , 1 , 2 , 3 , 4 , 5 , 6 , 7 , 8 } ) } , " " ) ; auto func = FunctionDefHelper :: FunctionRef ( " " , { { " " , DT_INT64 } } ) ; return FlatMapDatasetParams ( std :: move ( tensor_slice_dataset_params ) , { } , func , { test :: function :: NonZero ( ) } , { } , { DT_INT64 } , { PartialTensorShape ( { 1 } ) } , kNodeName ) ; }
Test case 2: test the case if the function does not return a single scalar
of dtype DT_VARIANT.
Test case 2 : test the case if the function does not return a single scalar of dtype DT_VARIANT .
/*components=*/ /*node_name=*/ /*name*/ /*attrs*/ /*other_arguments=*/ /*func=*/ /*func_lib=*/ /*type_arguments=*/ /*output_dtypes=*/ /*output_shapes=*/ /*node_name=*/