Searched refs:CheckOpKernelInput (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | dataset_test_base.h | 104 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, range_inputs)); in CreateRangeDataset() 140 Status CheckOpKernelInput(const OpKernel& kernel,
|
D | dataset_test_base.cc | 78 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_slice_dataset_kernel, inputs)); in CreateTensorSliceDataset() 220 Status DatasetOpsTestBase::CheckOpKernelInput( in CheckOpKernelInput() function in tensorflow::data::DatasetOpsTestBase
|
D | concatenate_dataset_op_test.cc | 63 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateConcatenateDatasetContext()
|
D | zip_dataset_op_test.cc | 73 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateZipDatasetContext()
|
D | range_dataset_op_test.cc | 52 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, inputs_)); in CreateRangeDatasetContext()
|
D | tensor_dataset_op_test.cc | 59 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_dataset_kernel, *inputs)); in CreateTensorDatasetContext()
|
D | sparse_tensor_slice_dataset_op_test.cc | 53 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateSparseTensorSliceDatasetContext()
|
D | take_dataset_op_test.cc | 52 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateTakeDatasetContext()
|
D | repeat_dataset_op_test.cc | 55 TF_RETURN_IF_ERROR(CheckOpKernelInput(*op_kernel, *inputs)); in CreateRepeatDatasetContext()
|
D | map_dataset_op_test.cc | 80 TF_RETURN_IF_ERROR(CheckOpKernelInput(*map_kernel, map_inputs_)); in CreateMapDatasetContext()
|
D | tensor_slice_dataset_op_test.cc | 61 TF_RETURN_IF_ERROR(CheckOpKernelInput(*tensor_dataset_kernel, *inputs)); in CreateTensorSliceDatasetContext()
|