Searched refs:input_tensor_index (Results 1 – 5 of 5) sorted by relevance
145 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local146 if (input_tensor_index != kOptionalTensor && in UpdateNode()147 tensor_epochs_[input_tensor_index] == kEpochNotReady) { in UpdateNode()170 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode() local171 if (input_tensor_index == kOptionalTensor) { in UpdateNode()174 int input_epoch = tensor_epochs_[input_tensor_index]; in UpdateNode()177 current_subset.input_tensors.push_back(input_tensor_index); in UpdateNode()183 input_subset.output_tensors.push_back(input_tensor_index); in UpdateNode()
197 Index<3> input_tensor_index;198 input_tensor_index[0] = output_tensor_index[sp0];199 input_tensor_index[1] = output_tensor_index[sp1];200 input_tensor_index[2] = output_tensor_index[sp2];202 int input_index = TensorIndexToFlat(input_tensor_index, input_dims);372 Index<NDIMS> input_tensor_index;373 input_tensor_index[0] = output_tensor_index[0]; // batch376 input_tensor_index[i] = output_tensor_index[i] - padding_left[i - 1];378 (input_tensor_index[i] >= 0 && input_tensor_index[i] < input_dims[i]);380 input_tensor_index[NDIMS - 1] = output_tensor_index[NDIMS - 1]; // channels[all …]
140 int input_tensor_index = interpreter->inputs()[0];142 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(input_tensor_index);
1180 int input_tensor_index = -1; in AddDequantizeOperatorsWhereNeeded() local1186 input_tensor_index = 0; in AddDequantizeOperatorsWhereNeeded()1196 int tensor_id = node->inputs->data[input_tensor_index]; in AddDequantizeOperatorsWhereNeeded()