Searched defs:use_weights (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/caffe2/perfkernels/ |
| H A D | hp_emblookup_codegen.py | 11 def unroll(uf, IndexType, InType, OutType, use_weights, isa, fused, use_offsets): argument
12 def compute(regid, InType, use_weights, isa, prefetch): argument
204 def generic(IndexType, InType, OutType, use_weights, isa, fused, use_offsets): argument
205 def compute(InType, use_weights, isa): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | count_ops.cc | 111 bool use_weights = weights.NumElements() > 0; in Compute() local
201 bool use_weights = weights.NumElements() > 0; in Compute() local
328 bool use_weights = weights.NumElements() > 0; in Compute() local
|