Searched refs:tensor_slice_dataset_kernel (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | tensor_slice_dataset_op_test.cc | 151 std::unique_ptr<OpKernel> tensor_slice_dataset_kernel; in TEST_F() local
153 &tensor_slice_dataset_kernel)); in TEST_F()
156 tensor_slice_dataset_kernel.get(), &inputs, in TEST_F()
159 TF_ASSERT_OK(CreateDataset(tensor_slice_dataset_kernel.get(), in TEST_F()
213 std::unique_ptr<OpKernel> tensor_slice_dataset_kernel; in TEST_F() local
215 &tensor_slice_dataset_kernel)); in TEST_F()
218 CreateTensorSliceDatasetContext(tensor_slice_dataset_kernel.get(), in TEST_F()
221 TF_ASSERT_OK(CreateDataset(tensor_slice_dataset_kernel.get(), in TEST_F()
250 std::unique_ptr<OpKernel> tensor_slice_dataset_kernel; in TEST_F() local
252 &tensor_slice_dataset_kernel)); in TEST_F()
[all …]
|
| D | dataset_test_base.cc | 41 std::unique_ptr<OpKernel>* tensor_slice_dataset_kernel) { in CreateTensorSliceDatasetKernel() argument
52 TF_RETURN_IF_ERROR(CreateOpKernel(node_def, tensor_slice_dataset_kernel)); in CreateTensorSliceDatasetKernel()
59 std::unique_ptr<OpKernel> tensor_slice_dataset_kernel; in CreateTensorSliceDataset() local
73 node_name, dtypes, shapes, &tensor_slice_dataset_kernel)); in CreateTensorSliceDataset()
78 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_slice_dataset_kernel, inputs)); in CreateTensorSliceDataset()
80 TF_RETURN_IF_ERROR(CreateOpKernelContext(tensor_slice_dataset_kernel.get(), in CreateTensorSliceDataset()
83 RunOpKernel(tensor_slice_dataset_kernel.get(), context.get())); in CreateTensorSliceDataset()
|
| D | dataset_test_base.h | 115 std::unique_ptr<OpKernel>* tensor_slice_dataset_kernel);
|