Home
last modified time | relevance | path

Searched refs:range_kernel (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Drange_dataset_op_test.cc40 int64 start, int64 end, int64 step, OpKernel* const range_kernel, in CreateRangeDatasetContext() argument
44 &inputs_, range_kernel->input_types(), TensorShape({}), {start})); in CreateRangeDatasetContext()
46 &inputs_, range_kernel->input_types(), TensorShape({}), {end})); in CreateRangeDatasetContext()
48 &inputs_, range_kernel->input_types(), TensorShape({}), {step})); in CreateRangeDatasetContext()
51 CreateOpKernelContext(range_kernel, &inputs_, range_context)); in CreateRangeDatasetContext()
52 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, inputs_)); in CreateRangeDatasetContext()
80 std::unique_ptr<OpKernel> range_kernel; in TEST_P() local
81 TF_ASSERT_OK(CreateRangeDatasetOpKernel<int64>("range", &range_kernel)); in TEST_P()
84 range_kernel.get(), &range_context)); in TEST_P()
87 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_P()
[all …]
Ddataset_test_base.h92 std::unique_ptr<OpKernel> range_kernel; in CreateRangeDataset() local
93 TF_RETURN_IF_ERROR(CreateRangeDatasetOpKernel<T>(node_name, &range_kernel)); in CreateRangeDataset()
96 &range_inputs, range_kernel->input_types(), TensorShape({}), {start})); in CreateRangeDataset()
98 &range_inputs, range_kernel->input_types(), TensorShape({}), {end})); in CreateRangeDataset()
100 &range_inputs, range_kernel->input_types(), TensorShape({}), {step})); in CreateRangeDataset()
102 TF_RETURN_IF_ERROR(CreateOpKernelContext(range_kernel.get(), &range_inputs, in CreateRangeDataset()
104 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, range_inputs)); in CreateRangeDataset()
105 TF_RETURN_IF_ERROR(RunOpKernel(range_kernel.get(), range_context.get())); in CreateRangeDataset()