Home
last modified time | relevance | path

Searched refs:CheckOpKernelInput (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h104 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, range_inputs)); in CreateRangeDataset()
140 Status CheckOpKernelInput(const OpKernel& kernel,
Ddataset_test_base.cc78 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_slice_dataset_kernel, inputs)); in CreateTensorSliceDataset()
220 Status DatasetOpsTestBase::CheckOpKernelInput( in CheckOpKernelInput() function in tensorflow::data::DatasetOpsTestBase
Dconcatenate_dataset_op_test.cc63 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateConcatenateDatasetContext()
Dzip_dataset_op_test.cc73 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateZipDatasetContext()
Drange_dataset_op_test.cc52 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, inputs_)); in CreateRangeDatasetContext()
Dtensor_dataset_op_test.cc59 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_dataset_kernel, *inputs)); in CreateTensorDatasetContext()
Dsparse_tensor_slice_dataset_op_test.cc53 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateSparseTensorSliceDatasetContext()
Dtake_dataset_op_test.cc52 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateTakeDatasetContext()
Drepeat_dataset_op_test.cc55 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateRepeatDatasetContext()
Dmap_dataset_op_test.cc80 TF_RETURN_IF_ERROR(CheckOpKernelInput(*map_kernel, map_inputs_)); in CreateMapDatasetContext()
Dtensor_slice_dataset_op_test.cc61 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_dataset_kernel, *inputs)); in CreateTensorSliceDatasetContext()