Searched defs:squeezeDims (Results 1 – 9 of 9) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| H A D | squeeze_float16.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
35 squeezeDims = Parameter("squeezeDims", ["TENSOR_INT32", [0]], value=None) variable
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/ |
| H A D | squeeze_quant8_signed.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
33 squeezeDims = Parameter("squeezeDims", ["TENSOR_INT32", [0]], value=None) variable
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| H A D | squeeze_omitted.mod.py | 18 squeezeDims = Parameter("squeezeDims", ["TENSOR_INT32", [0]], value=None) variable
|
| H A D | squeeze.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
|
| H A D | squeeze_float_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
|
| H A D | squeeze_quant8_1.mod.py | 3 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
|
| H A D | squeeze_float_1_relaxed.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{1}", [2]) variable
|
| H A D | squeeze_relaxed.mod.py | 19 squeezeDims = Parameter("squeezeDims", "TENSOR_INT32", "{2}", [1, 2]) variable
|
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Squeeze.cpp | 75 const int32_t* squeezeDims = context->getInputBuffer<int32_t>(kSqueezeDims); in prepare() local
|