Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc191 OpInputList dense_float_features_list; in DoCompute() local
193 context, &dense_float_features_list)); in DoCompute()
215 dense_float_features_list, sparse_float_feature_shapes_list, in DoCompute()
223 TensorUtils::OpInputListToTensorVec(dense_float_features_list), in DoCompute()
403 OpInputList dense_float_features_list; in DoCompute() local
405 context, &dense_float_features_list)); in DoCompute()
427 dense_float_features_list, sparse_float_feature_shapes_list, in DoCompute()
435 TensorUtils::OpInputListToTensorVec(dense_float_features_list), in DoCompute()
Dquantile_ops.cc360 OpInputList dense_float_features_list; in Compute() local
362 context, &dense_float_features_list)); in Compute()
410 dense_float_features_list[dense_index].flat<float>(); in Compute()
774 OpInputList dense_float_features_list; in Compute() local
776 context, &dense_float_features_list)); in Compute()
830 dense_float_features_list[dense_index].flat<float>(); in Compute()
880 OpInputList dense_float_features_list; in Compute() local
882 &dense_float_features_list)); in Compute()
919 QuantizeFeatures(kDenseOutputTensorName, dense_float_features_list, in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dtensor_utils.cc86 const OpInputList& dense_float_features_list, in InferBatchSize() argument
89 if (dense_float_features_list.size() > 0) { in InferBatchSize()
90 return dense_float_features_list[0].dim_size(0); in InferBatchSize()
Dbatch_features.cc26 std::vector<Tensor> dense_float_features_list, in Initialize() argument
34 auto num_dense_float_features = dense_float_features_list.size(); in Initialize()
46 auto dense_float_feature = dense_float_features_list[dense_feat_idx]; in Initialize()
Dtensor_utils.h51 const OpInputList& dense_float_features_list,
Dbatch_features.h40 Status Initialize(std::vector<Tensor> dense_float_features_list,
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Dbatch_features_testutil.cc33 std::vector<tensorflow::Tensor> dense_float_features_list; in RandomlyInitializeBatchFeatures() local
41 dense_float_features_list.push_back(dense_tensor); in RandomlyInitializeBatchFeatures()
81 dense_float_features_list, sparse_float_feature_indices_list, in RandomlyInitializeBatchFeatures()