Searched refs:cross_count (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/contrib/layers/kernels/ |
H A D | sparse_feature_cross_kernel.cc | 148 void Update(const int64 batch_index, const int64 cross_count, argument 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; 154 indices_matrix(output_index, 1) = cross_count; 376 int64 cross_count = 0; 379 updater.Update(b, cross_count, crosser.Generate(b, permutation)); 380 cross_count++; 558 const auto cross_count = CrossCountByBatchIndex(columns, b); local 559 max_cross_count = std::max(max_cross_count, cross_count); 560 cross_count_total += cross_count; 584 int64 cross_count local [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
H A D | sparse_cross_op.cc | 148 void Update(const int64 batch_index, const int64 cross_count, argument 150 const int64 output_index = output_start_indices_[batch_index] + cross_count; 154 indices_matrix(output_index, 1) = cross_count; 333 int64 cross_count = 0; 336 updater.Update(b, cross_count, crosser.Generate(b, permutation)); 337 cross_count++; 515 const auto cross_count = CrossCountByBatchIndex(columns, b); local 516 max_cross_count = std::max(max_cross_count, cross_count); 517 cross_count_total += cross_count; 541 int64 cross_count local [all...] |
Completed in 258 milliseconds