Searched defs:n_inputs (Results 1 – 8 of 8) sorted by relevance
189 int n_inputs = network->getNbInputs(); in SetShapeTensorMask() local285 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles() local
74 int n_inputs = network->getNbInputs(); in SetDimensions() local
140 int n_inputs = ctx ? ctx->num_inputs() : (input_vec ? input_vec->size() : 0); in SetTrtEngineInputs() local
60 int n_inputs = node_->input_size(); in ParseInputs() local
72 const int n_inputs = op_inputs.size(); in Init() local
187 const int n_inputs = op->Inputs().size(); in ValidateInputTypeAndPlacement() local
186 const int n_inputs = network->getNbInputs(); in GetNetworkInputShapes() local
112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad() local