Searched refs:input_index_features (Results 1 – 2 of 2) sorted by relevance
125 TensorView<float> ComputeLogitsHelper(const int input_index_features, in ComputeLogitsHelper() argument132 interpreter->ResizeInputTensor(input_index_features, features.shape()); in ComputeLogitsHelper()139 interpreter->tensor(interpreter->inputs()[input_index_features]); in ComputeLogitsHelper()
40 TensorView<float> ComputeLogitsHelper(const int input_index_features,