Home
last modified time | relevance | path

Searched refs:kSampleCountParam (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DMultinomial.cpp57 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
74 getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Prepare()
DMultinomial.h43 static constexpr int kSampleCountParam = 1; variable
DMultinomialTest.cpp74 ASSERT_EQ(execution.setInput(Multinomial::kSampleCountParam, &sample_size_, in Invoke()