Searched refs:range_context (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op_test.cc | 41 std::unique_ptr<OpKernelContext>* range_context) { in CreateRangeDatasetContext() argument
51 CreateOpKernelContext(range_kernel, &inputs_, range_context)); in CreateRangeDatasetContext()
82 std::unique_ptr<OpKernelContext> range_context; in TEST_P() local
84 range_kernel.get(), &range_context)); in TEST_P()
87 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_P()
91 TF_ASSERT_OK(CreateIteratorContext(range_context.get(), &iterator_context)); in TEST_P()
131 std::unique_ptr<OpKernelContext> range_context; in TEST_F() local
133 &range_context)); in TEST_F()
136 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_F()
151 std::unique_ptr<OpKernelContext> range_context; in TEST_F() local
[all …]
|
| D | dataset_test_base.h | 101 std::unique_ptr<OpKernelContext> range_context; in CreateRangeDataset() local
103 &range_context)); in CreateRangeDataset()
105 TF_RETURN_IF_ERROR(RunOpKernel(range_kernel.get(), range_context.get())); in CreateRangeDataset()
107 GetDatasetFromContext(range_context.get(), 0, range_dataset)); in CreateRangeDataset()
|