Searched refs:features_tensor (Results 1 – 2 of 2) sorted by relevance
268 features_tensor = input_fn()273 feature = session.run(features_tensor)278 session.run([features_tensor])280 session.run([features_tensor])340 features_tensor, targets_tensor = input_fn()345 features, targets = session.run([features_tensor, targets_tensor])353 session.run([features_tensor, targets_tensor])355 session.run([features_tensor, targets_tensor])422 features_tensor, targets_tensor = input_fn()425 features, targets = session.run([features_tensor, targets_tensor])[all …]
138 TfLiteTensor* features_tensor = in ComputeLogitsHelper() local141 for (int i = 0; i < features_tensor->dims->size; ++i) { in ComputeLogitsHelper()142 size *= features_tensor->dims->data[i]; in ComputeLogitsHelper()144 features.copy_to(features_tensor->data.f, size); in ComputeLogitsHelper()