Searched refs:cross_count (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/layers/kernels/
H A Dsparse_feature_cross_kernel.cc148 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 Dsparse_cross_op.cc148 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