Home
last modified time | relevance | path

Searched refs:ResultType (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/common/random/
Drandom_distributions.h81 typedef Array<Eigen::half, kResultElementCount> ResultType; typedef
85 ResultType operator()(Generator* gen) { in operator()
86 typename Generator::ResultType sample = (*gen)(); in operator()
87 ResultType result; in operator()
105 typedef Array<bfloat16, kResultElementCount> ResultType; typedef
109 ResultType operator()(Generator* gen) { in operator()
110 typename Generator::ResultType sample = (*gen)(); in operator()
111 ResultType result; in operator()
129 typedef Array<float, kResultElementCount> ResultType; typedef
133 ResultType operator()(Generator* gen) { in operator()
[all …]
Dphilox_random.h107 using ResultType = Array<uint32, 4>;
135 PhiloxRandom(ResultType counter, Key key) : counter_(counter), key_(key) {} in PhiloxRandom()
158 PHILOX_DEVICE_INLINE ResultType operator()() { in operator()
159 ResultType counter = counter_; in operator()
222 PHILOX_DEVICE_INLINE static ResultType ComputeSingleRound(const ResultType& counter, in ComputeSingleRound()
232 ResultType result; in ComputeSingleRound()
246 ResultType counter_;
Dguarded_philox_random.cc33 void GuardedPhiloxRandom::Init(random::PhiloxRandom::ResultType counter, in Init()
Dguarded_philox_random.h51 void Init(random::PhiloxRandom::ResultType counter, random::PhiloxRandom::Key key);
/frameworks/compile/slang/
Dslang_rs_reflection.cpp563 for (const RSExportType *ResultType : mRSContext->getReduceResultTypes( in genScriptClass() local
570 genExportReduceResultType(ResultType); in genScriptClass()
1224 bool RSReflectionJava::exportableReduce(const RSExportType *ResultType) { in exportableReduce() argument
1225 const RSExportType *CheckType = ResultType; in exportableReduce()
1226 if (ResultType->getClass() == RSExportType::ExportClassConstantArray) in exportableReduce()
1227 CheckType = static_cast<const RSExportConstantArrayType *>(ResultType)->getElementType(); in exportableReduce()
1308 const RSExportType *const ResultType = ER->getResultType(); in genExportReduceArrayVariant() local
1309 auto ResultTypeClass = ResultType->getClass(); in genExportReduceArrayVariant()
1331 ResultType->convertToRTD(&ResultTypeData); in genExportReduceArrayVariant()
1461 const RSExportType *ResultType = ER->getResultType(); in genExportReduceAllocationVariant() local
[all …]
Dslang_rs_reflection.h209 static bool exportableReduce(const RSExportType *ResultType);
243 void genExportReduceResultType(const RSExportType *ResultType);
Dslang_rs_reflection_state.cpp899 const std::string ResultType = getUniqueTypeName(ER->getResultType()); in declareReduce() local
900 if (!reduce.mResult.equals(ResultType)) { in declareReduce()
904 << reduce.mName << reduce.mResult.str() << ResultType; in declareReduce()