Searched refs:SaveNewAxisTensor (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | strided_slice_builder_test.cpp | 39 void SaveNewAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim,
87 void StridedSliceBuilderTest::SaveNewAxisTensor(OH_NN_DataType dataType, in SaveNewAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::StridedSliceBuilderTest
127 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK); in InitParams()
267 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
288 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
309 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
330 SaveNewAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
351 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
378 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
405 SaveNewAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STRIDED_SLICE_NEW_AXIS_MASK);
[all …]
|