HomeSort by relevance Sort by last modified time
    Searched refs:indices_matrix (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/data/experimental/
dense_to_sparse_batch_dataset_op.cc 237 auto indices_matrix = indices.matrix<int64>(); variable
258 indices_matrix(current_position_in_values, 0) = i;
261 indices_matrix(current_position_in_values, k + 1) =
  /external/tensorflow/tensorflow/core/kernels/
sparse_cross_op.cc 152 auto indices_matrix = indices_out_->matrix<int64>(); local
153 indices_matrix(output_index, 0) = batch_index;
154 indices_matrix(output_index, 1) = cross_count;
  /external/tensorflow/tensorflow/contrib/layers/kernels/
sparse_feature_cross_kernel.cc 152 auto indices_matrix = indices_out_->matrix<int64>(); local
153 indices_matrix(output_index, 0) = batch_index;
154 indices_matrix(output_index, 1) = cross_count;

Completed in 841 milliseconds