| /aosp_15_r20/external/pytorch/torch/utils/bottleneck/ |
| H A D | __main__.py | 88 def print_cprofile_summary(prof, sortby='tottime', topk=15): argument
120 def print_autograd_prof_summary(prof, mode, sortby='cpu_time', topk=15): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | sort_ops_test.py | 254 def topk(v, k=k): function
290 def topk(v, k=k): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | topk_ops.cc | 219 auto topk = CreateTopKUnique(builder, input, input_shape, k_, false); in Compile() local
349 auto topk = CreateTopKUnique(builder, unique, input_shape, k_, true); in Compile() local
|
| /aosp_15_r20/external/pytorch/android/test_app/app/src/main/java/org/pytorch/testapp/ |
| H A D | Utils.java | 7 public static int[] topK(float[] a, final int topk) { in topK()
|
| /aosp_15_r20/test/mlts/benchmark/src/com/android/nn/benchmark/evaluators/ |
| D | TopK.java | 50 int[] topk = new int[K_TOP]; in EvaluateAccuracy() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/experimental/ |
| H A D | quantization_util.py | 55 def accuracy(output, target, topk=(1,)): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | topk_rewriter.cc | 233 HloInstruction* topk = comp->AddInstruction( in TransformToCustomCall() local
|
| H A D | buffer_assignment.cc | 783 absl::btree_multimap<int64_t, const HloValue*> topk; in TopKPeakBuffers() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | custom_call_handler.cc | 133 auto topk = b_.AddInstruction( in HandleCustomCallTopK() local
|
| H A D | spmd_partitioner.cc | 2501 auto topk = b_.AddInstruction(hlo->CloneWithNewOperands( in HandleSort() local
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchvision_models.py | 643 def accuracy(output, target, topk=(1,)): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Sorting.cpp | 54 TORCH_META_FUNC(topk) in TORCH_META_FUNC() argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset10.py | 121 def topk(g: jit_utils.GraphContext, self, k, dim, largest, sorted, out=None): function
|
| H A D | symbolic_opset11.py | 556 def topk(g: jit_utils.GraphContext, self, k, dim, largest, sorted, out=None): function
|
| H A D | symbolic_opset9.py | 3920 def topk(g: jit_utils.GraphContext, self, k, dim, largest, sorted, out=None): function
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1090 def topk(self: List[int], k: int, dim: int = -1) -> Tuple[List[int], List[int]]: function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 177 def accuracy(output, target, topk=(1,)): argument
|
| /aosp_15_r20/external/python/google-api-python-client/googleapiclient/discovery_cache/documents/ |
| D | apigee.v1.json | 4857 "topk": { object
5554 "topk": { object
6093 "topk": { object
6822 "topk": { object
8643 "topk": { object
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 7008 auto topk = ops::TopK(s.WithOpName("my_topk"), input, weights); in TEST_P() local
|