Searched refs:blockShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1982 ANeuralNetworksOperandType blockShape = {.type = ANEURALNETWORKS_TENSOR_INT32, in spaceBatchOpTest() local
1993 ANeuralNetworksOperandType padding = blockShape; in spaceBatchOpTest()
1996 OperationTestBase spaceBatchTest(operationCode, {input, blockShape, padding}, {output}); in spaceBatchOpTest()
1999 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, padding, layout}, in spaceBatchOpTest()
2003 OperationTestBase spaceBatchTest(operationCode, {input, blockShape}, {output}); in spaceBatchOpTest()
2006 OperationTestBase spaceBatchNchwTest(operationCode, {input, blockShape, layout}, {output}); in spaceBatchOpTest()
|