Searched refs:tensor_inputs (Results 1 – 3 of 3) sorted by relevance
70 tensor_inputs = []73 tensor_inputs.append(75 …result = function._call_flat(tensor_inputs, function._captured_inputs) # pylint: disable=protecte…
1900 tensor_inputs = []1909 tensor_inputs.append(arg.handle)1912 tensor_inputs.append(arg)1937 args = tensor_inputs + captured_inputs
6199 std::vector<nvinfer1::ITensor*> tensor_inputs; in ConvertAddN() local6202 tensor_inputs.push_back(input.tensor()); in ConvertAddN()6206 tensor_inputs.push_back( in ConvertAddN()6210 nvinfer1::ITensor* lhs = tensor_inputs[0]; in ConvertAddN()6212 nvinfer1::ITensor* rhs = tensor_inputs[i]; in ConvertAddN()