Searched refs:batch_key (Results 1 – 3 of 3) sorted by relevance
| /external/linux-kselftest/tools/testing/selftests/bpf/map_tests/ |
| D | map_in_map_batch_ops.c | 126 __u32 batch_key = 0, fetch_count, step_size; in fetch_and_validate() local
143 total_fetched ? &batch_key : NULL, in fetch_and_validate()
144 &batch_key, in fetch_and_validate()
149 total_fetched ? &batch_key : NULL, in fetch_and_validate()
150 &batch_key, in fetch_and_validate()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 664 const int64_t batch_key = context->input(2).scalar<int64_t>()(); in Compute() local
690 auto tensor_it = waiting_tensors_.find(batch_key); in Compute()
703 .emplace(batch_key, in Compute()
903 const int64_t batch_key = context->input(3).scalar<int64_t>()(); in Compute() local
905 if (!available_tensors_.emplace(batch_key, grad_t).second) { in Compute()
925 const int64_t batch_key = batch_index(i, 0); in Compute() local
926 if (available_tensors_.find(batch_key) == available_tensors_.end()) { in Compute()
927 missing_tensors.emplace(batch_key); in Compute()
934 .emplace(batch_key, Batch{missing_tensors, context, done}) in Compute()
940 if (!desired_tensor_to_batch_map_.emplace(i, batch_key).second) { in Compute()
[all …]
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | group_batch_fusion.py | 215 batch_key = ("batch_linear_post_grad", m, k, n, bias_m is not None, str(users))
216 return batch_key
|