tensorflow/core/kernels/data/flat_map_dataset_op_test.cc
// Test case 1: normal case.
Test case 1: normal case.
Test case 1 : normal case .
FlatMapDatasetParams FlatMapDatasetParams1() {
  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(
      "MakeTensorSliceDataset",
      {{"Toutput_types", DataTypeVector({DT_INT64})},
                 {"output_shapes",
                  std::vector<PartialTensorShape>({PartialTensorShape({1})})}});
  return FlatMapDatasetParams(
      std::move(tensor_slice_dataset_params),
      {},
      func,
      {test::function::MakeTensorSliceDataset()},
      {},
      {DT_INT64},
      {PartialTensorShape({1})},
      kNodeName);
}
FlatMapDatasetParams FlatMapDatasetParams1 ( ) { 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 ( " " , { { " " , DataTypeVector ( { DT_INT64 } ) } , { " " , std :: vector < PartialTensorShape > ( { PartialTensorShape ( { 1 } ) } ) } } ) ; return FlatMapDatasetParams ( std :: move ( tensor_slice_dataset_params ) , { } , func , { test :: function :: MakeTensorSliceDataset ( ) } , { } , { DT_INT64 } , { PartialTensorShape ( { 1 } ) } , kNodeName ) ; }
Test case 1: normal case.
Test case 1 : normal case .
/*components=*/ /*node_name=*/ /*name=*/ /*attrs=*/ /*other_arguments=*/ /*func=*/ /*func_lib=*/ /*type_arguments=*/ /*output_dtypes=*/ /*output_shapes=*/ /*node_name=*/