Searched refs:kLayoutScalar (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ResizeImageOps.cpp | 39 constexpr uint32_t kLayoutScalar = 3; variable
147 if (context->getNumInputs() > kLayoutScalar) { in prepare()
148 useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
196 if (context->getNumInputs() > kLayoutScalar) { in execute()
197 useNchw = context->getInputValue<bool>(kLayoutScalar); in execute()
|
| D | InstanceNormalization.cpp | 38 constexpr uint32_t kLayoutScalar = 4; variable
126 context->getInputValue<bool>(kLayoutScalar), in execute()
135 context->getInputValue<bool>(kLayoutScalar), in execute()
|
| D | RoiPooling.cpp | 40 constexpr uint32_t kLayoutScalar = 7; variable
203 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
260 context->getInputValue<bool>(kLayoutScalar), in execute()
272 context->getInputValue<bool>(kLayoutScalar), in execute()
284 context->getInputValue<bool>(kLayoutScalar), in execute()
|
| D | RoiAlign.cpp | 43 constexpr uint32_t kLayoutScalar = 9; variable
368 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
440 context->getInputValue<bool>(kLayoutScalar), in execute()
454 context->getInputValue<bool>(kLayoutScalar), in execute()
468 context->getInputValue<bool>(kLayoutScalar), in execute()
|
| D | HeatmapMaxKeypoint.cpp | 35 constexpr uint32_t kLayoutScalar = 2; variable
225 bool layout = context->getInputValue<bool>(kLayoutScalar); in prepare()
260 bool layout = context->getInputValue<bool>(kLayoutScalar); in execute()
|
| D | GenerateProposals.cpp | 815 constexpr uint32_t kLayoutScalar = 10; variable
1165 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
1236 context->getInputValue<bool>(kLayoutScalar), context); in execute()
1253 context->getInputValue<bool>(kLayoutScalar), context); in execute()
1270 context->getInputValue<bool>(kLayoutScalar), context); in execute()
|