Searched refs:sparse_index (Results 1 – 2 of 2) sorted by relevance
421 const int64 sparse_index = i - num_dense_features_; in Compute() local423 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() local808 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 …]
142 const int32 sparse_index = in RandomSample() local145 (*sparse_indices_)(sparse_index, 1) + input_spec_.dense_features_size(); in RandomSample()