Searched defs:n_inputs (Results 1 – 15 of 15) sorted by relevance
420 const nvinfer1::ICudaEngine* engine, int n_inputs) { in SetShapeTensorMask()440 int n_inputs = network->getNbInputs(); in SetShapeTensorMask() local598 int n_inputs = GetNumberOfEngineInputs(engine); in RestoreProfiles() local
124 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
213 const int n_inputs = op->Inputs().size(); in ValidateInputTypeAndPlacement() local
143 const int n_inputs = network->getNbInputs(); in GetNetworkInputShapes() local
3383 int n_inputs = 2) -> NodeDef { in TEST_P()3601 int n_inputs = p.shape_b.empty() ? 1 : 2; in TEST_P() local
1335 int n_inputs = network()->getNbInputs(); in BuildCudaEngine() local
62 int n_inputs; member
112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad() local
341 Status ValidateConversionParams(const TfTrtConversionParams& p, int n_inputs) { in ValidateConversionParams()
2188 const auto n_inputs = static_cast<uint32_t>(inputs_.size()); in verify_inputs_dont_overlap_outputs() local2227 const auto n_inputs = static_cast<uint32_t>(inputs_.size()); in verify_and_correct_memory_overlap() local
190 const size_t n_inputs = input_tf_tensors.size(); in ValidateInputTypes() local
104 size_t n_inputs; member
3679 int n_inputs = inputs_ty.size(); in verify() local3735 int n_inputs = inputs_ty.size(); in verify() local