/external/tensorflow/tensorflow/core/framework/ |
D | op_compatibility_test.cc | 54 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectSuccess() 62 TF_ASSERT_OK(ValidateNodeDef(*node_def(), *new_op_def)); in ExpectSuccess() 64 TF_ASSERT_OK(InOutTypesForNode(*node_def(), *new_op_def, &new_in_types, in ExpectSuccess() 84 TF_ASSERT_OK(OpDefCompatible(old_op_def, *new_op_def)); in ExpectSuccess() 87 TF_ASSERT_OK(InitOp()); in ExpectSuccess() 88 TF_ASSERT_OK(RunOpKernel()); in ExpectSuccess() 110 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectInvalid() 133 TF_ASSERT_OK(InOutTypesForNode(*node_def(), old_op_def, &old_in_types, in ExpectTypeMismatch() 141 TF_ASSERT_OK(ValidateNodeDef(*node_def(), *new_op_def)); in ExpectTypeMismatch() 144 TF_ASSERT_OK(InOutTypesForNode(*node_def(), *new_op_def, &new_in_types, in ExpectTypeMismatch() [all …]
|
D | graph_def_util_test.cc | 41 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("Project>SomeOp"), op_list.add_op())); in TEST() 45 TF_ASSERT_OK(NodeDefBuilder("node", "Project>SomeOp", ®istry) in TEST() 52 TF_ASSERT_OK( in TEST() 58 TF_ASSERT_OK(NodeDefBuilder("ncwd", "NoChangeWithDefault", ®istry) in TEST() 63 TF_ASSERT_OK(RemoveNewDefaultAttrsFromGraphDef(&graph_def, registry, registry, in TEST() 73 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("NoChangeNoDefault").Attr("a: int"), in TEST() 78 TF_ASSERT_OK(NodeDefBuilder("ncnd", "NoChangeNoDefault", ®istry) in TEST() 84 TF_ASSERT_OK(RemoveNewDefaultAttrsFromGraphDef(&graph_def, registry, registry, in TEST() 96 TF_ASSERT_OK( in TEST() 101 TF_ASSERT_OK(FinalizeOpDef(OpDefBuilder("UsesDefault").Attr("a: int = 17"), in TEST() [all …]
|
/external/tensorflow/tensorflow/core/lib/io/ |
D | buffered_inputstream_test.cc | 68 TF_ASSERT_OK(WriteStringToFile(env, fname, "")); in TEST() 70 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file)); in TEST() 85 TF_ASSERT_OK( in TEST() 88 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file)); in TEST() 95 TF_ASSERT_OK(in.ReadLine(&line)); in TEST() 97 TF_ASSERT_OK(in.ReadLine(&line)); in TEST() 99 TF_ASSERT_OK(in.ReadLine(&line)); in TEST() 111 TF_ASSERT_OK(WriteStringToFile(env, fname, "line one\nline two\nline three")); in TEST() 113 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file)); in TEST() 120 TF_ASSERT_OK(in.ReadLine(&line)); in TEST() [all …]
|
D | random_inputstream_test.cc | 29 TF_ASSERT_OK(WriteStringToFile(env, fname, "0123456789")); in TEST() 32 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file)); in TEST() 35 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST() 38 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST() 41 TF_ASSERT_OK(in.ReadNBytes(5, &read)); in TEST() 44 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST() 50 TF_ASSERT_OK(in.ReadNBytes(0, &read)); in TEST() 59 TF_ASSERT_OK(WriteStringToFile(env, fname, "0123456789")); in TEST() 62 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file)); in TEST() 67 TF_ASSERT_OK(in.ReadNBytes(3, &read)); in TEST() [all …]
|
D | zlib_buffers_test.cc | 74 TF_ASSERT_OK(env->NewWritableFile(fname, &file_writer)); in TestAllCombinations() 79 TF_ASSERT_OK(out.Init()); in TestAllCombinations() 81 TF_ASSERT_OK(out.Append(StringPiece(data))); in TestAllCombinations() 82 TF_ASSERT_OK(out.Close()); in TestAllCombinations() 83 TF_ASSERT_OK(file_writer->Flush()); in TestAllCombinations() 84 TF_ASSERT_OK(file_writer->Close()); in TestAllCombinations() 87 TF_ASSERT_OK(env->NewRandomAccessFile(fname, &file_reader)); in TestAllCombinations() 92 TF_ASSERT_OK(in.ReadNBytes(data.size(), &result)); in TestAllCombinations() 125 TF_ASSERT_OK(env->NewWritableFile(fname, &file_writer)); in TestMultipleWrites() 128 TF_ASSERT_OK(out.Init()); in TestMultipleWrites() [all …]
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | shape_refiner_test.cc | 80 TF_ASSERT_OK(NodeBuilder("test", "TensorAsShapeInt32") in TestStridedSlice() 85 TF_ASSERT_OK(m.AddNode(placeholder.node())); in TestStridedSlice() 86 TF_ASSERT_OK(m.AddNode(input.node())); in TestStridedSlice() 87 TF_ASSERT_OK(m.AddNode(begin_op.node())); in TestStridedSlice() 88 TF_ASSERT_OK(m.AddNode(end_op.node())); in TestStridedSlice() 89 TF_ASSERT_OK(m.AddNode(stride_op.node())); in TestStridedSlice() 90 TF_ASSERT_OK(m.AddNode(slice.node())); in TestStridedSlice() 91 TF_ASSERT_OK(m.AddNode(result)); in TestStridedSlice() 130 TF_ASSERT_OK(m.AddNode(c.node())); in TEST_F() 143 TF_ASSERT_OK(m.AddNode(a.node())); in TEST_F() [all …]
|
D | quantize_training_test.cc | 107 TF_ASSERT_OK(DoQuantizeTraining(num_bits, "QuantizeAndDequantizeV2", g)); in TEST_F() 113 TF_ASSERT_OK( in TEST_F() 120 TF_ASSERT_OK( in TEST_F() 158 TF_ASSERT_OK(DoQuantizeTraining(num_bits, "QuantizeAndDequantizeV2", g)); in TEST_F() 164 TF_ASSERT_OK( in TEST_F() 171 TF_ASSERT_OK( in TEST_F() 202 TF_ASSERT_OK(NodeBuilder(g->NewName("gradients/n"), "MatMul") in TEST_F() 211 TF_ASSERT_OK(DoQuantizeTraining(num_bits, "QuantizeAndDequantizeV2", g)); in TEST_F() 222 TF_ASSERT_OK( in TEST_F() 225 TF_ASSERT_OK( in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/lib/db/ |
D | sqlite_test.cc | 32 TF_ASSERT_OK(Sqlite::Open(":memory:", SQLITE_OPEN_READWRITE, &db_)); in SetUp() 46 TF_ASSERT_OK(stmt.StepAndReset()); in TEST_F() 49 TF_ASSERT_OK(stmt.StepAndReset()); in TEST_F() 51 TF_ASSERT_OK(stmt.Step(&is_done_)); in TEST_F() 55 TF_ASSERT_OK(stmt.Step(&is_done_)); in TEST_F() 59 TF_ASSERT_OK(stmt.Step(&is_done_)); in TEST_F() 67 TF_ASSERT_OK(stmt.StepAndReset()); in TEST_F() 69 TF_ASSERT_OK(stmt.Step(&is_done_)); in TEST_F() 83 TF_ASSERT_OK(stmt.StepAndReset()); in TEST_F() 85 TF_ASSERT_OK(stmt.Step(&is_done_)); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantize_and_dequantize_op_test.cc | 47 TF_ASSERT_OK( in TEST_F() 56 TF_ASSERT_OK(InitOp()); in TEST_F() 61 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 72 TF_ASSERT_OK( in TEST_F() 81 TF_ASSERT_OK(InitOp()); in TEST_F() 87 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 124 TF_ASSERT_OK( in TEST_P() 134 TF_ASSERT_OK(InitOp()); in TEST_P() 154 TF_ASSERT_OK(RunOpKernel()); in TEST_P() 174 TF_ASSERT_OK( in TEST_P() [all …]
|
D | debug_ops_test.cc | 63 TF_ASSERT_OK(Init(DT_INT32)); in TEST_F() 65 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 88 TF_ASSERT_OK(Init(DT_INT32, debug_urls)); in TEST_F() 90 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 102 TF_ASSERT_OK(Env::Default()->GetFileSystemForFile(dump_roots[i], &fs)); in TEST_F() 104 TF_ASSERT_OK(fs->GetChildren(dump_roots[i], &children)); in TEST_F() 117 TF_ASSERT_OK(Env::Default()->GetFileSystemForFile(device_root, &fs)); in TEST_F() 118 TF_ASSERT_OK(fs->GetChildren(device_root, &children)); in TEST_F() 161 TF_ASSERT_OK(Init(DT_INT32)); in TEST_F() 163 TF_ASSERT_OK(RunOpKernel()); in TEST_F() [all …]
|
D | remote_fused_graph_execute_utils_test.cc | 69 TF_ASSERT_OK( in SetUp() 226 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildAddGraph( in TEST() 247 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildAddGraph( in TEST() 267 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildAddGraph( in TEST() 301 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildAddGraph( in TEST() 344 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildAddGraph( in TEST() 377 TF_ASSERT_OK( in TEST() 381 TF_ASSERT_OK(GetNodeAttr( in TEST() 388 TF_ASSERT_OK(GetNodeAttr( in TEST() 396 TF_ASSERT_OK( in TEST() [all …]
|
D | as_string_op_test.cc | 49 TF_ASSERT_OK(Init(DT_INT8)); in TEST_F() 52 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 59 TF_ASSERT_OK(Init(DT_INT64)); in TEST_F() 62 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 69 TF_ASSERT_OK(Init(DT_FLOAT)); in TEST_F() 72 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 80 TF_ASSERT_OK(Init(DT_FLOAT, /*fill=*/"", /*width=*/-1, /*precision=*/-1, in TEST_F() 84 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 92 TF_ASSERT_OK(Init(DT_FLOAT, /*fill=*/"", /*width=*/-1, /*precision=*/-1, in TEST_F() 96 TF_ASSERT_OK(RunOpKernel()); in TEST_F() [all …]
|
D | remote_fused_graph_rewriter_transform_test.cc | 75 TF_ASSERT_OK(RemoteFusedGraphExecuteOpTestUtils::BuildMultipleAddGraph( in SetUp() 190 TF_ASSERT_OK( in CheckGraph() 222 TF_ASSERT_OK(Fuse()); in TEST_F() 229 TF_ASSERT_OK(Fuse()); in TEST_F() 237 TF_ASSERT_OK(Fuse()); in TEST_F() 244 TF_ASSERT_OK(Fuse()); in TEST_F() 253 TF_ASSERT_OK(Fuse()); in TEST_F() 261 TF_ASSERT_OK(Fuse()); in TEST_F() 270 TF_ASSERT_OK(Fuse()); in TEST_F() 278 TF_ASSERT_OK(Fuse()); in TEST_F() [all …]
|
D | quantize_op_test.cc | 38 TF_ASSERT_OK(NodeDefBuilder("quantize_op", "QuantizeV2") in TEST_F() 45 TF_ASSERT_OK(InitOp()); in TEST_F() 52 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 87 TF_ASSERT_OK(NodeDefBuilder("quantize_op", "QuantizeV2") in TEST_P() 95 TF_ASSERT_OK(InitOp()); in TEST_P() 111 TF_ASSERT_OK(RunOpKernel()); in TEST_P() 136 TF_ASSERT_OK(NodeDefBuilder("quantize_op", "QuantizeV2") in TEST_F() 143 TF_ASSERT_OK(InitOp()); in TEST_F() 147 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 167 TF_ASSERT_OK(NodeDefBuilder("quantize_op", "QuantizeV2") in TEST_P() [all …]
|
D | fingerprint_op_test.cc | 67 TF_ASSERT_OK(MakeFingerprintOp(&tensor)); in TEST_F() 68 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 80 TF_ASSERT_OK(MakeFingerprintOp(&tensor)); in TEST_F() 81 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 100 TF_ASSERT_OK(MakeFingerprintOp(&data)); in TEST_F() 101 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 108 TF_ASSERT_OK(MakeFingerprintOp(&data)); in TEST_F() 109 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 127 TF_ASSERT_OK(MakeFingerprintOp(&tensor)); in TEST_F() 128 TF_ASSERT_OK(RunOpKernel()); in TEST_F() [all …]
|
D | logging_ops_test.cc | 45 TF_ASSERT_OK(Init()); in TEST_F() 47 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 55 TF_ASSERT_OK(Init()); in TEST_F() 76 TF_ASSERT_OK(Init(DT_INT32, DT_INT32)); in TEST_F() 80 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 87 TF_ASSERT_OK(Init(DT_INT32, DT_INT32, "", -1, 6)); in TEST_F() 91 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 98 TF_ASSERT_OK(Init(DT_INT32, DT_STRING)); in TEST_F() 102 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 109 TF_ASSERT_OK(Init(DT_INT32, DT_STRING, "Message: ")); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/kernels/batching_util/ |
D | shared_batch_scheduler_test.cc | 95 TF_ASSERT_OK( in TEST() 104 TF_ASSERT_OK( in TEST() 107 TF_ASSERT_OK( in TEST() 117 TF_ASSERT_OK(ScheduleTask(1, queue_0.get())); in TEST() 118 TF_ASSERT_OK(ScheduleTask(2, queue_1.get())); in TEST() 119 TF_ASSERT_OK(ScheduleTask(3, queue_0.get())); in TEST() 120 TF_ASSERT_OK(ScheduleTask(4, queue_1.get())); in TEST() 124 TF_ASSERT_OK(ScheduleTask(5, queue_0.get())); in TEST() 156 TF_ASSERT_OK(SharedBatchScheduler<FakeTask>::Create(options, &scheduler)); in TEST() 162 TF_ASSERT_OK(scheduler->AddQueue(queue_options, callback, &queue)); in TEST() [all …]
|
D | adaptive_shared_batch_scheduler_test.cc | 125 TF_ASSERT_OK( in TEST() 128 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue)); in TEST() 131 TF_ASSERT_OK(ScheduleTask(100, queue.get())); in TEST() 134 TF_ASSERT_OK(ScheduleTask(100, queue.get())); in TEST() 137 TF_ASSERT_OK(ScheduleTask(100, queue.get())); in TEST() 168 TF_ASSERT_OK( in TEST() 171 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue)); in TEST() 173 TF_ASSERT_OK(ScheduleTask(0, queue.get())); in TEST() 180 TF_ASSERT_OK(ScheduleTask(1, queue.get())); in TEST() 186 TF_ASSERT_OK(ScheduleTask(2, queue.get())); in TEST() [all …]
|
D | serial_device_batch_scheduler_test.cc | 119 TF_ASSERT_OK( in TEST() 124 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue1)); in TEST() 125 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue2)); in TEST() 126 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue3)); in TEST() 128 TF_ASSERT_OK(ScheduleTask(100, queue1.get())); in TEST() 129 TF_ASSERT_OK(ScheduleTask(100, queue2.get())); in TEST() 130 TF_ASSERT_OK(ScheduleTask(100, queue3.get())); in TEST() 146 TF_ASSERT_OK( in TEST() 188 TF_ASSERT_OK(scheduler->AddQueue({}, queue_callback, &queue)); in TEST() 191 TF_ASSERT_OK(ScheduleTask(800, queue.get())); in TEST() [all …]
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
D | trt_engine_resource_ops_test.cc | 116 TF_ASSERT_OK(NodeDefBuilder("op", "CreateTRTResourceHandle") in TEST_F() 119 TF_ASSERT_OK(InitOp()); in TEST_F() 120 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 135 TF_ASSERT_OK(env->NewWritableFile(filename, &file)); in TEST_F() 137 TF_ASSERT_OK(NodeDefBuilder("op", "InitializeTRTResource") in TEST_F() 142 TF_ASSERT_OK(InitOp()); in TEST_F() 145 TF_ASSERT_OK(RunOpKernel()); in TEST_F() 156 TF_ASSERT_OK(context_status.status()); in TEST_F() 169 TF_ASSERT_OK(NodeDefBuilder("op", "SerializeTRTResource") in TEST_F() 174 TF_ASSERT_OK(InitOp()); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | map_dataset_op_test.cc | 110 TF_ASSERT_OK(Initialize(dataset_params)); in ITERATOR_GET_NEXT_TEST_P() 111 TF_ASSERT_OK(CheckDatasetNodeName(dataset_params.node_name())); in ITERATOR_GET_NEXT_TEST_P() 116 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 117 TF_ASSERT_OK( in TEST_F() 123 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 124 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F() 129 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 130 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F() 147 TF_ASSERT_OK(Initialize(dataset_params)); in DATASET_CARDINALITY_TEST_P() 148 TF_ASSERT_OK(CheckIteratorOutputDtypes({DT_INT64})); in DATASET_CARDINALITY_TEST_P() [all …]
|
D | flat_map_dataset_op_test.cc | 135 TF_ASSERT_OK(Initialize(dataset_params)); in ITERATOR_GET_NEXT_TEST_P() 136 TF_ASSERT_OK(CheckDatasetNodeName(dataset_params.node_name())); in ITERATOR_GET_NEXT_TEST_P() 141 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 142 TF_ASSERT_OK(CheckDatasetTypeString( in TEST_F() 148 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 149 TF_ASSERT_OK(CheckDatasetOutputDtypes(dataset_params.output_dtypes())); in TEST_F() 154 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 155 TF_ASSERT_OK(CheckDatasetOutputShapes(dataset_params.output_shapes())); in TEST_F() 160 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 161 TF_ASSERT_OK(CheckDatasetCardinality(kUnknownCardinality)); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | fold_batch_norms_test.cc | 67 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsConv2D() 70 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsConv2D() 72 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsConv2D() 75 TF_ASSERT_OK( in TestFoldBatchNormsConv2D() 79 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsConv2D() 81 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldBatchNormsConv2D() 118 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative() 121 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative() 123 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsDepthwiseConv2dNative() 126 TF_ASSERT_OK( in TestFoldBatchNormsDepthwiseConv2dNative() [all …]
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties_test.cc | 65 TF_ASSERT_OK(cluster_->Provision()); in SetUp() 93 TF_ASSERT_OK(cluster_->Shutdown()); in TearDown() 161 TF_ASSERT_OK(s); in TEST_F() 202 TF_ASSERT_OK(s); in TEST_F() 228 TF_ASSERT_OK(cluster_->Initialize(item)); in TEST_F() 230 TF_ASSERT_OK(s); in TEST_F() 280 TF_ASSERT_OK(NodeDefBuilder("Var", "Variable") in TEST_F() 288 TF_ASSERT_OK(NodeDefBuilder("InitialVal", "Const") in TEST_F() 292 TF_ASSERT_OK(NodeDefBuilder("InitVar", "Assign") in TEST_F() 300 TF_ASSERT_OK(static_properties.InferStatically(false)); in TEST_F() [all …]
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | assert_next_dataset_op_test.cc | 146 TF_ASSERT_OK(Initialize(dataset_params)); in ITERATOR_GET_NEXT_TEST_P() 147 TF_ASSERT_OK(CheckDatasetNodeName(dataset_params.node_name())); in ITERATOR_GET_NEXT_TEST_P() 152 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 153 TF_ASSERT_OK(CheckDatasetTypeString( in TEST_F() 159 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 160 TF_ASSERT_OK(CheckDatasetOutputDtypes({DT_INT64})); in TEST_F() 165 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 166 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F() 171 TF_ASSERT_OK(Initialize(dataset_params)); in TEST_F() 172 TF_ASSERT_OK(CheckDatasetCardinality(/*expected_cardinality=*/3)); in TEST_F() [all …]
|