Searched refs:float_features_list (Results 1 – 1 of 1) sorted by relevance
147 OpInputList float_features_list; in Compute() local149 context, context->input_list(kFloatFeaturesName, &float_features_list)); in Compute()155 DCHECK(float_features_list.size() > 0) << "Got empty feature list"; in Compute()158 const int64 batch_size = float_features_list[0].flat<float>().size(); in Compute()174 const auto feature_values = float_features_list[index].flat<float>(); in Compute()451 OpInputList float_features_list; in Compute() local453 context, context->input_list(kFloatFeaturesName, &float_features_list)); in Compute()468 const Tensor& values_tensor = float_features_list[feature_idx]; in Compute()