Searched defs:n_inputs (Results 1 – 12 of 12) sorted by relevance
363 const nvinfer1::ICudaEngine* engine, int n_inputs) { in SetShapeTensorMask()378 int n_inputs = network->getNbInputs(); in SetShapeTensorMask() local511 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles() local
78 int n_inputs = network->getNbInputs(); in SetDimensions() local
103 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
203 const int n_inputs = op->Inputs().size(); in ValidateInputTypeAndPlacement() local
60 int n_inputs; member
172 const int n_inputs = network->getNbInputs(); in GetNetworkInputShapes() local
2769 int n_inputs = 2) -> NodeDef { in TEST_P()2940 int n_inputs = p.shape_b.empty() ? 1 : 2; in TEST_P() local
1546 int n_inputs = network()->getNbInputs(); in BuildCudaEngine() local
112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad() local
170 const size_t n_inputs = input_tf_tensors.size(); in ValidateInputTypes() local