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