Searched refs:selector_input (Results 1 – 2 of 2) sorted by relevance
41 Dataset(OpKernelContext* ctx, const DatasetBase* selector_input, in Dataset() argument44 selector_input_(selector_input), in Dataset()285 DatasetBase* selector_input; in MakeDataset() local287 GetDatasetFromVariantTensor(ctx->input(0), &selector_input)); in MakeDataset()291 selector_input->output_dtypes().size() == 1 && in MakeDataset()292 selector_input->output_dtypes()[0] == DT_INT64 && in MakeDataset()293 selector_input->output_shapes().size() == 1 && in MakeDataset()294 selector_input->output_shapes()[0].IsCompatibleWith( in MakeDataset()313 *output = new Dataset(ctx, selector_input, std::move(data_inputs)); in MakeDataset()
107 def __init__(self, selector_input, data_inputs): argument108 self._selector_input = selector_input207 selector_input = dataset_ops.MapDataset(227 selector_input = dataset_ops.MapDataset(232 return _DirectedInterleaveDataset(selector_input, datasets)