Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h102 TF_RETURN_IF_ERROR(CreateOpKernelContext(range_kernel.get(), &range_inputs, in CreateRangeDataset()
144 Status CreateOpKernelContext(OpKernel* kernel,
Ddataset_test_base.cc80 TF_RETURN_IF_ERROR(CreateOpKernelContext(tensor_slice_dataset_kernel.get(), in CreateTensorSliceDataset()
178 Status DatasetOpsTestBase::CreateOpKernelContext( in CreateOpKernelContext() function in tensorflow::data::DatasetOpsTestBase
Dconcatenate_dataset_op_test.cc64 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateConcatenateDatasetContext()
Dzip_dataset_op_test.cc74 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateZipDatasetContext()
Drange_dataset_op_test.cc51 CreateOpKernelContext(range_kernel, &inputs_, range_context)); in CreateRangeDatasetContext()
Dtensor_dataset_op_test.cc61 CreateOpKernelContext(tensor_dataset_kernel, inputs, context)); in CreateTensorDatasetContext()
Dsparse_tensor_slice_dataset_op_test.cc54 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateSparseTensorSliceDatasetContext()
Dtake_dataset_op_test.cc53 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateTakeDatasetContext()
Drepeat_dataset_op_test.cc56 TF_RETURN_IF_ERROR(CreateOpKernelContext(op_kernel, inputs, context)); in CreateRepeatDatasetContext()
Dmap_dataset_op_test.cc79 CreateOpKernelContext(map_kernel, &map_inputs_, map_context)); in CreateMapDatasetContext()
Dtensor_slice_dataset_op_test.cc63 CreateOpKernelContext(tensor_dataset_kernel, inputs, context)); in CreateTensorSliceDatasetContext()