Home
last modified time | relevance | path

Searched refs:expand_dims (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/test/generated/tests/
Dexpand_dims.mod.py.cpp5 namespace expand_dims { namespace
12 TEST_F(GeneratedTests, expand_dims) { in TEST_F() argument
13 execute(expand_dims::CreateModel, in TEST_F()
14 expand_dims::is_ignored, in TEST_F()
15 expand_dims::get_examples()); in TEST_F()
17 TEST_AVAILABLE_SINCE(V1_2, expand_dims, expand_dims::CreateModel) in TEST_AVAILABLE_SINCE() argument
20 execute(expand_dims::CreateModel_relaxed, in TEST_AVAILABLE_SINCE()
21 expand_dims::is_ignored_relaxed, in TEST_AVAILABLE_SINCE()
22 expand_dims::get_examples_relaxed()); in TEST_AVAILABLE_SINCE()
26 execute(expand_dims::CreateModel_quant8, in TEST_F()
[all …]
/frameworks/ml/nn/common/operations/
DExpandDims.h24 namespace expand_dims {
DExpandDims.cpp25 namespace expand_dims { namespace
/frameworks/ml/nn/common/
DCpuExecutor.cpp1457 success = expand_dims::prepare(input.shape(), axis, &outShape) && in executeOperation()
1459 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()