Home
last modified time | relevance | path

Searched refs:sparse_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc421 const int64 sparse_index = i - num_dense_features_; in Compute() local
423 sparse_float_feature_values_list[sparse_index].flat<float>(); in Compute()
425 sparse_float_feature_indices_list[sparse_index].matrix<int64>(); in Compute()
427 sparse_float_feature_shapes_list[sparse_index].flat<int64>(); in Compute()
434 sparse_float_feature_indices_list[sparse_index].dim_size(0); in Compute()
441 copy_over_summaries(stream, sparse_index, in Compute()
806 const int64 sparse_index = i; in Compute() local
808 sparse_float_feature_values_list[sparse_index].flat<float>(); in Compute()
810 sparse_float_feature_indices_list[sparse_index].matrix<int64>(); in Compute()
811 QuantileStream stream(sparse_configs_[sparse_index].eps(), in Compute()
[all …]
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dinput_data.cc142 const int32 sparse_index = in RandomSample() local
145 (*sparse_indices_)(sparse_index, 1) + input_spec_.dense_features_size(); in RandomSample()