Searched refs:num_to_sample_tensor (Results 1 – 1 of 1) sorted by relevance
85 const Tensor& num_to_sample_tensor = context->input(1); in Compute() local92 TensorShapeUtils::IsScalar(num_to_sample_tensor.shape()), in Compute()103 const int64_t num_to_sample = num_to_sample_tensor.scalar<int64_t>()(); in Compute()