| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | candidate_sampling_ops.py | 32 def uniform_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
95 def log_uniform_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
158 def learned_unigram_candidate_sampler(true_classes, num_true, num_sampled, argument
221 num_true, argument
312 def all_candidate_sampler(true_classes, num_true, num_sampled, unique, argument
349 def compute_accidental_hits(true_classes, sampled_candidates, num_true, argument
|
| H A D | nn_impl.py | 1831 num_true=1, argument
2018 num_true=1, argument
2123 num_true=1, argument
2235 num_true=1, argument
2327 num_true=1, argument
|
| H A D | nn_test.py | 527 def _GenerateTestData(self, num_classes, dim, batch_size, num_true, labels, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UniqueCub.cu | 202 const int num_true = *num_true_p; in C10_LAUNCH_BOUNDS_1() local
219 const int num_true = *num_true_p; in unique_bool_write_output() local
292 int num_true = 0; in operator ()() local
|
| H A D | TensorModeKernel.cu | 110 const auto num_true = thrust::count_if( in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | candidate_sampling_ops.cc | 30 int64_t num_true; in CandidateSamplerShapeFn() local
145 int64_t num_true; in __anonda2a69000202() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | where_op.cc | 143 int64_t num_true; in Compute() local
282 ScratchSpace<Tindex> num_true(context, 1, /*on_host=*/true); in ComputeAsyncType() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | advanced_index_util.cpp | 90 size_t num_true = _count_trues_in_mask_index<CTYPE_IX>(index); in _query_mask_index() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unique.cpp | 68 int64_t num_true = std::accumulate(num_true_thread.begin(), num_true_thread.end(), 0); in unique_cpu_bool_template() local
|