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 Dbatch_features_testutil.cc47 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 Dbatch_features.cc28 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 Dtensor_utils.cc87 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 Dprediction_ops.cc190 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 Dquantile_ops.cc359 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 125 milliseconds