Searched refs:ds_tensor (Results 1 – 1 of 1) sorted by relevance
86 explicit WrappedDatasetVariantWrapper(const Tensor& ds_tensor) in WrappedDatasetVariantWrapper() argument87 : ds_tensor_(ds_tensor) {} in WrappedDatasetVariantWrapper()154 Tensor ds_tensor = wrapper->get(); in Compute() local155 OP_REQUIRES_OK(ctx, ctx->set_output("output_handle", ds_tensor)); in Compute()