Searched refs:num_nonzero_elements (Results 1 – 2 of 2) sorted by relevance
94 const int64 num_nonzero_elements = mask_indices.dim_size(0); in Compute() local110 num_nonzero_elements, 2); in Compute()114 0, TensorShape({num_nonzero_elements}), in Compute()117 num_nonzero_elements); in Compute()149 std::vector<int64> perm(num_nonzero_elements); in Compute()184 worker_threads.workers->ParallelFor(num_nonzero_elements, cost_per_unit, in Compute()
124 const int64 num_nonzero_elements = input_indices.dim_size(0); in Compute() local135 2, num_nonzero_elements); in Compute()166 std::vector<int64> perm(num_nonzero_elements); in Compute()185 while (end < num_nonzero_elements) { in Compute()187 while (end < num_nonzero_elements && in Compute()194 CHECK_EQ(shard_total, num_nonzero_elements); in Compute()195 CHECK_LE(shards.size(), num_nonzero_elements); in Compute()