Home
last modified time | relevance | path

Searched refs:sparse_float_feature_shapes_list (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc198 OpInputList sparse_float_feature_shapes_list; in DoCompute() local
202 &sparse_float_feature_shapes_list)); in DoCompute()
215 dense_float_features_list, sparse_float_feature_shapes_list, in DoCompute()
229 sparse_float_feature_shapes_list), in DoCompute()
410 OpInputList sparse_float_feature_shapes_list; in DoCompute() local
414 &sparse_float_feature_shapes_list)); in DoCompute()
427 dense_float_features_list, sparse_float_feature_shapes_list, in DoCompute()
441 sparse_float_feature_shapes_list), in DoCompute()
Dquantile_ops.cc367 OpInputList sparse_float_feature_shapes_list; in Compute() local
371 &sparse_float_feature_shapes_list)); in Compute()
427 sparse_float_feature_shapes_list[sparse_index].flat<int64>(); in Compute()
781 OpInputList sparse_float_feature_shapes_list; in Compute() local
785 &sparse_float_feature_shapes_list)); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dtensor_utils.cc87 const OpInputList& sparse_float_feature_shapes_list, in InferBatchSize() argument
92 if (sparse_float_feature_shapes_list.size() > 0) { in InferBatchSize()
93 return sparse_float_feature_shapes_list[0].flat<int64>()(0); in InferBatchSize()
Dbatch_features.cc29 std::vector<Tensor> sparse_float_feature_shapes_list, in Initialize() argument
67 sparse_float_feature_shapes_list.size() == num_sparse_float_features, in Initialize()
76 sparse_float_feature_shapes_list[sparse_feat_idx]; in Initialize()
Dtensor_utils.h52 const OpInputList& sparse_float_feature_shapes_list,
Dbatch_features.h43 std::vector<Tensor> sparse_float_feature_shapes_list,
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Dbatch_features_testutil.cc47 std::vector<tensorflow::Tensor> sparse_float_feature_shapes_list; in RandomlyInitializeBatchFeatures() local
76 sparse_float_feature_shapes_list.push_back(shape_tensor); in RandomlyInitializeBatchFeatures()
82 sparse_float_feature_values_list, sparse_float_feature_shapes_list, {}, in RandomlyInitializeBatchFeatures()