Home
last modified time | relevance | path

Searched refs:IsScalar (Results 1 – 5 of 5) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dlayernorm_builder.cpp44 if (!tensor->IsScalar()) { in SetBeginNormAxis()
67 if (!tensor->IsScalar()) { in SetEpsilon()
90 if (!tensor->IsScalar()) { in SetBeginParamsAxis()
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnn_tensor.h78 bool IsScalar() const;
H A Dnn_tensor.cpp477 bool NNTensor::IsScalar() const in IsScalar() function in OHOS::NeuralNetworkRuntime::NNTensor
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/inner_model/
H A Dnn_tensor_test.cpp305 EXPECT_EQ(false, nnTensor.IsScalar());
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v2_0/inner_model/
H A Dnn_tensor_test.cpp305 EXPECT_EQ(false, nnTensor.IsScalar());