Searched refs:expand_dims (Results 1 – 4 of 4) sorted by relevance
/frameworks/ml/nn/runtime/test/generated/tests/ |
D | expand_dims.mod.py.cpp | 5 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/ |
D | ExpandDims.h | 24 namespace expand_dims {
|
D | ExpandDims.cpp | 25 namespace expand_dims { namespace
|
/frameworks/ml/nn/common/ |
D | CpuExecutor.cpp | 1457 success = expand_dims::prepare(input.shape(), axis, &outShape) && in executeOperation() 1459 expand_dims::eval(input.buffer, input.shape(), axis, output.buffer, outShape); in executeOperation()
|