Searched refs:kSampleCountParam (Results 1 – 3 of 3) sorted by relevance
57 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()74 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
43 static constexpr int kSampleCountParam = 1; variable
74 ASSERT_EQ(execution.setInput(Multinomial::kSampleCountParam, &sample_size_, in Invoke()