Searched refs:blockShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1981 ANeuralNetworksOperandType blockShape = {.type = ANEURALNETWORKS_TENSOR_INT32, in spaceBatchOpTest() local
1992 ANeuralNetworksOperandType padding = blockShape; in spaceBatchOpTest()
1995 OperationTestBase spaceBatchTest(operationCode, {input, blockShape, padding}, {output}); in spaceBatchOpTest()
1998 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, padding, layout}, in spaceBatchOpTest()
2002 OperationTestBase spaceBatchTest(operationCode, {input, blockShape}, {output}); in spaceBatchOpTest()
2005 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, layout}, {output}); in spaceBatchOpTest()
|