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

  /external/tensorflow/tensorflow/core/kernels/
sparse_add_grad_op.cc 70 const int64 sum_nnz = backprop_val_grad->NumElements(); variable
107 while (i < a_nnz && j < b_nnz && k < sum_nnz) {
117 while (i < a_nnz && k < sum_nnz) {
122 while (j < b_nnz && k < sum_nnz) {
sparse_sparse_binary_op_shared.cc 193 const int64 sum_nnz = a_augmented_values.size(); variable
196 ctx->allocate_output(0, TensorShape({sum_nnz, num_dims}),
199 ctx, ctx->allocate_output(1, TensorShape({sum_nnz}), &output_values_t));
202 for (int64 i = 0; i < sum_nnz; ++i) {
220 UnalignedTensorMap(a_augmented_values.data(), sum_nnz);
222 UnalignedTensorMap(b_augmented_values.data(), sum_nnz);
sparse_add_op.cc 146 const int64 sum_nnz = out_values.size(); variable
149 ctx->allocate_output(0, TensorShape({sum_nnz, num_dims}),
152 ctx, ctx->allocate_output(1, TensorShape({sum_nnz}), &out_values_t));
156 for (i = 0; i < sum_nnz; ++i) {
162 std::copy_n(out_values.begin(), sum_nnz, &out_values_flat(0));
sparse_conditional_accumulator.h 216 int64 sum_nnz = 0; variable
218 sum_nnz++;
238 sum_nnz++;
243 sum_nnz++;
250 sum_indices_vec->reserve(sum_nnz);
253 sum_counts->reserve(sum_nnz);
259 sum_shape.set_dim(0, sum_nnz);
272 for (i = 0; i < sum_nnz; ++i) {

Completed in 760 milliseconds