Searched refs:weight_shape (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | EmbeddingLookupTest.cpp | 58 std::initializer_list<uint32_t> weight_shape) { in EmbeddingLookupOpModel() argument
59 auto it = weight_shape.begin(); in EmbeddingLookupOpModel()
69 OperandType ValueTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel()
74 OperandType OutputOpndTy(Type::TENSOR_FLOAT32, weight_shape); in EmbeddingLookupOpModel()
85 Value_.insert(Value_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()
86 Output_.insert(Output_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()
|
| D | LSHProjectionTest.cpp | 46 std::initializer_list<uint32_t> weight_shape) in LSHProjectionOpModel() argument
54 OperandType WeightTy(Type::TENSOR_FLOAT32, weight_shape); in LSHProjectionOpModel()
|
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.cpp | 65 const Shape& weight_shape, const int32_t* bias_tensor, in PrecomputeZeroPointTimesWeightWithBias() argument
71 NN_RET_CHECK_EQ(weight_shape.dimensions.size(), 2u); in PrecomputeZeroPointTimesWeightWithBias()
72 const int row = weight_shape.dimensions[0]; in PrecomputeZeroPointTimesWeightWithBias()
73 const int col = weight_shape.dimensions[1]; in PrecomputeZeroPointTimesWeightWithBias()
|
| D | QuantUtils.h | 144 const Shape& weight_shape, const int32_t* bias_tensor,
|