Searched refs:BATCH_MATMUL (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_AIDL_V2/ |
| D | batch_matmul.example.cpp | 64 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple()
209 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_all_inputs_as_internal()
286 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_float16()
431 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_float16_all_inputs_as_internal()
508 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_int32()
585 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_quant8_signed()
730 .type = TestOperationType::BATCH_MATMUL, in get_test_model_Simple_quant8_signed_all_inputs_as_internal()
807 .type = TestOperationType::BATCH_MATMUL, in get_test_model_RHSAdjoint()
952 .type = TestOperationType::BATCH_MATMUL, in get_test_model_RHSAdjoint_all_inputs_as_internal()
1029 .type = TestOperationType::BATCH_MATMUL, in get_test_model_RHSAdjoint_float16()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | BatchMatmul.cpp | 56 NN_DEFINE_VALIDATION_FUNCTION(BATCH_MATMUL, batch_matmul_op::validate);
|
| /packages/modules/NeuralNetworks/common/types/include/nnapi/ |
| D | OperationTypes.h | 5265 BATCH_MATMUL = 102, enumerator
|
| /packages/modules/NeuralNetworks/runtime/test/android_fuzzing/ |
| D | StaticAssert.cpp | 176 static_assert(static_cast<TestOperationType>(BATCH_MATMUL) == TestOperationType::BATCH_MATMUL);
|
| D | Model.proto | 143 BATCH_MATMUL = 102; enumerator
|
| /packages/modules/NeuralNetworks/runtime/operation_converters/ |
| D | OperationConverterResolver.cpp | 132 NN_OPERATION_CONVERTER_NOT_IMPLEMENTED(BATCH_MATMUL);
|
| /packages/modules/NeuralNetworks/common/types/include/ |
| D | OperationsUtils.h | 145 CALL(BATCH_MATMUL) \
|
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BatchMatmul.cpp | 276 NN_REGISTER_OPERATION_DEFAULT_VALIDATION(BATCH_MATMUL, batch_matmul_op::prepare,
|
| /packages/modules/NeuralNetworks/tools/test_generator/test_harness/include/ |
| D | TestHarness.h | 197 BATCH_MATMUL = 102, enumerator
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | NeuralNetworksV2.cpp | 558 static_assert(static_cast<int32_t>(OperationType::BATCH_MATMUL) == ANEURALNETWORKS_BATCH_MATMUL,
|
| D | NeuralNetworks.cpp | 556 static_assert(static_cast<int32_t>(OperationType::BATCH_MATMUL) == ANEURALNETWORKS_BATCH_MATMUL,
|