Searched refs:tensorHasUnspecifiedDimensions (Results 1 – 6 of 6) sorted by relevance
/frameworks/ml/nn/common/ |
D | BufferTracker.cpp | 135 if (tensorHasUnspecifiedDimensions(kOperandType, kInitialDimensions)) { in validateCopyFrom() 142 if (tensorHasUnspecifiedDimensions(kOperandType, dimensions)) { in validateCopyFrom()
|
D | UtilsTest.cpp | 70 EXPECT_TRUE(tensorHasUnspecifiedDimensions(getExtensionType(1, 0), /*dim=*/nullptr, in TEST()
|
D | Utils.cpp | 407 bool tensorHasUnspecifiedDimensions(int type, const uint32_t* dim, uint32_t dimCount) { in tensorHasUnspecifiedDimensions() function 415 bool tensorHasUnspecifiedDimensions(OperandType type, const std::vector<uint32_t>& dimensions) { in tensorHasUnspecifiedDimensions() function 416 return tensorHasUnspecifiedDimensions(static_cast<int>(type), dimensions.data(), in tensorHasUnspecifiedDimensions() 420 bool tensorHasUnspecifiedDimensions(const ANeuralNetworksOperandType* type) { in tensorHasUnspecifiedDimensions() function 421 return tensorHasUnspecifiedDimensions(type->type, type->dimensions, type->dimensionCount); in tensorHasUnspecifiedDimensions() 424 bool tensorHasUnspecifiedDimensions(const Operand& operand) { in tensorHasUnspecifiedDimensions() function 425 return tensorHasUnspecifiedDimensions(static_cast<int>(operand.type), operand.dimensions.data(), in tensorHasUnspecifiedDimensions()
|
/frameworks/ml/nn/common/include/ |
D | Utils.h | 311 bool tensorHasUnspecifiedDimensions(int type, const uint32_t* dim, uint32_t dimCount); 312 bool tensorHasUnspecifiedDimensions(hal::OperandType type, const std::vector<uint32_t>& dimensions); 313 bool tensorHasUnspecifiedDimensions(const hal::Operand& operand); 314 bool tensorHasUnspecifiedDimensions(const ANeuralNetworksOperandType* type);
|
/frameworks/ml/nn/runtime/ |
D | ModelBuilder.cpp | 130 tensorHasUnspecifiedDimensions(operand)) { in setOperandValue() 325 if (TypeManager::get()->isTensorType(operand.type) && tensorHasUnspecifiedDimensions(operand)) { in setOperandValueFromMemory()
|
D | ExecutionBuilder.cpp | 82 tensorHasUnspecifiedDimensions(operand)) { in checkDimensionInfo()
|