Searched defs:sparse_float_feature_shapes_list (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
H A D | batch_features_testutil.cc | 47 std::vector<tensorflow::Tensor> sparse_float_feature_shapes_list; local 76 sparse_float_feature_shapes_list.push_back(shape_tensor); 82 sparse_float_feature_values_list, sparse_float_feature_shapes_list, {},
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
H A D | batch_features.cc | 28 std::vector<Tensor> sparse_float_feature_shapes_list, 66 sparse_float_feature_shapes_list.size() == num_sparse_float_features, 75 sparse_float_feature_shapes_list[sparse_feat_idx]; 24 Initialize( std::vector<Tensor> dense_float_features_list, std::vector<Tensor> sparse_float_feature_indices_list, std::vector<Tensor> sparse_float_feature_values_list, std::vector<Tensor> sparse_float_feature_shapes_list, std::vector<Tensor> sparse_int_feature_indices_list, std::vector<Tensor> sparse_int_feature_values_list, std::vector<Tensor> sparse_int_feature_shapes_list) argument
|
H A D | tensor_utils.cc | 87 const OpInputList& sparse_float_feature_shapes_list, 92 if (sparse_float_feature_shapes_list.size() > 0) { 93 return sparse_float_feature_shapes_list[0].flat<int64>()(0); 85 InferBatchSize( const OpInputList& dense_float_features_list, const OpInputList& sparse_float_feature_shapes_list, const OpInputList& sparse_int_feature_shapes_list) argument
|
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
H A D | prediction_ops.cc | 190 OpInputList sparse_float_feature_shapes_list; local 194 &sparse_float_feature_shapes_list)); 207 dense_float_features_list, sparse_float_feature_shapes_list, 221 sparse_float_feature_shapes_list), 372 OpInputList sparse_float_feature_shapes_list; local 376 &sparse_float_feature_shapes_list)); 389 dense_float_features_list, sparse_float_feature_shapes_list, 403 sparse_float_feature_shapes_list),
|
H A D | quantile_ops.cc | 359 OpInputList sparse_float_feature_shapes_list; variable 363 &sparse_float_feature_shapes_list)); 419 sparse_float_feature_shapes_list[sparse_index].flat<int64>(); 772 OpInputList sparse_float_feature_shapes_list; variable 776 &sparse_float_feature_shapes_list));
|
Completed in 260 milliseconds