Searched refs:GetDimensions (Results 1 – 8 of 8) sorted by relevance
145 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()182 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in ValidateGammaAndBetaShape()183 auto gammaShape = allTensors[inputsIndex[INPUT_GAMMA]]->GetDimensions(); in ValidateGammaAndBetaShape()184 auto betaShape = allTensors[inputsIndex[INPUT_BETA]]->GetDimensions(); in ValidateGammaAndBetaShape()
45 auto blockshape_shape = tensor->GetDimensions(); in SetBlockShape()78 auto paddings_shape = tensor->GetDimensions(); in SetPaddings()
91 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()220 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build()
63 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetChannel()79 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
57 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetInput()72 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize()
69 std::vector<int32_t> GetDimensions() const;
409 std::vector<int32_t> NNTensor::GetDimensions() const in GetDimensions() function in OHOS::NeuralNetworkRuntime::NNTensor499 const std::vector<int32_t> dimensions = tensor.GetDimensions(); in CompareAttribute()
404 const std::vector<int32_t> dimensions = tensor.GetDimensions(); in CompareAttribute()