Searched refs:samples_t (Results 1 – 3 of 3) sorted by relevance
181 Tensor* samples_t; in DoCompute() local184 &samples_t)); in DoCompute()187 if (samples_t->NumElements() > 0) { in DoCompute()215 samples_t->matrix<OutputType>()); in DoCompute()
298 Tensor* samples_t = nullptr; in Compute() local299 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, samples_shape, &samples_t)); in Compute()304 auto samples_flat = samples_t->flat<U>().data(); in Compute()
299 Tensor* samples_t = nullptr; in Compute() local300 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, samples_shape, &samples_t)); in Compute()325 auto samples_flat = samples_t->flat<T>().data(); in Compute()