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