| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_to_dense_op.cc | 61 const int64_t num_elems = indices.dims() > 0 ? indices.dim_size(0) : 1; in CheckSparseToDenseShapes() local
111 const int64_t num_elems = indices.dims() > 0 ? indices.dim_size(0) : 1; in Compute() local
|
| H A D | sparse_to_dense_op_gpu.cu.cc | 113 const T* values, const int num_elems, in LaunchComputeKernels()
155 const int64 num_elems = indices.dims() > 0 ? indices.dim_size(0) : 1; in operator ()() local
|
| H A D | sparse_reorder_op_gpu.cu.cc | 74 const int64 num_elems = input_ind.dims() > 0 ? input_ind.dim_size(0) : 1; in operator ()() local
|
| H A D | tensor_to_hash_bucket_op_gpu.cu.cc | 89 const int num_elems, int64* output) { in operator ()()
|
| /aosp_15_r20/external/sg3_utils/testing/ |
| H A D | sg_scat_gath.cpp | 52 scat_gath_list::num_elems() const in num_elems() function in scat_gath_list
665 const int num_elems = sgl.size(); in get_lowest_lba() local
691 const int num_elems = sgl.size(); in get_low_lba_from_linear() local
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | program.cpp | 295 size_t num_elems = internal_program->constant_segment()->offsets()->size(); in get_constant_buffer_data() local
327 size_t num_elems = internal_program->constant_buffer()->size(); in get_constant_buffer_data() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | prng.cc | 391 std::pair<Philox4x32State, XlaOp> GeneratePhiloxBits(int64_t num_elems, in GeneratePhiloxBits()
411 const int64_t num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit32() local
438 const int64_t num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit64() local
|
| /aosp_15_r20/frameworks/native/libs/binder/tests/ |
| H A D | binderThroughputTest.cpp | 153 size_t num_elems = v.data.size(); in send() local
169 size_t num_elems = 0; in recv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | ctc_ops.py | 1427 def cond(i, num_elems, *args): argument
1433 def body(i, num_elems, *args): argument
|
| /aosp_15_r20/external/rappor/client/cpp/ |
| H A D | unix_kernel_rand_impl.cc | 25 size_t num_elems = fread(&rand_buf, sizeof(uint8_t), num_bits, fp_); in GetMask() local
|
| /aosp_15_r20/external/trusty/lk/lib/libc/ |
| D | bsearch.c | 17 void *bsearch(const void *key, const void *base, size_t num_elems, size_t size, in bsearch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| H A D | tpu_driver.cc | 104 int64_t num_elems = 1; in ComputeBytesFromShape() local
|
| /aosp_15_r20/external/libnetfilter_conntrack/src/conntrack/ |
| H A D | stack.c | 18 int num_elems; member
|
| /aosp_15_r20/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/crypto/test/ |
| D | malloc.cc | 123 void *calloc(size_t num_elems, size_t size) { in calloc()
|
| /aosp_15_r20/external/rust/crates/quiche/deps/boringssl/src/crypto/test/ |
| D | malloc.cc | 123 void *calloc(size_t num_elems, size_t size) { in calloc()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qtanh.cpp | 39 size_t num_elems = 1; in qnnpack_tanh() local
|
| H A D | qhardsigmoid.cpp | 37 size_t num_elems = input_contig.numel() / input_contig.size(0); in qnnpack_hardsigmoid() local
|
| H A D | qhardswish.cpp | 35 size_t num_elems = qx.numel() / qx.size(0); in qnnpack_hardswish() local
|
| H A D | qsigmoid.cpp | 42 size_t num_elems = 1; in qnnpack_sigmoid() local
|
| /aosp_15_r20/external/flatbuffers/java/com/google/flatbuffers/ |
| H A D | FlatBufferBuilder.java | 492 public void startVector(int elem_size, int num_elems, int alignment) { in startVector()
526 public ByteBuffer createUnintializedVector(int elem_size, int num_elems, int alignment) { in createUnintializedVector()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sparse_to_dense_op.cc | 37 const int64_t num_elems = in Compile() local
|
| /aosp_15_r20/external/pdfium/core/fxcrt/ |
| H A D | fx_memory.cpp | 25 void* FXMEM_DefaultCalloc(size_t num_elems, size_t byte_size) { in FXMEM_DefaultCalloc()
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/multi_index/test/ |
| D | test_update.cpp | 59 size_type num_elems= in test_stable_update() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/remat/ |
| H A D | metadata_util.cc | 89 uint32_t num_elems = 0; in Parse() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/ |
| H A D | init.cpp | 80 auto num_elems = arr.size(); in getTupleAttr() local
|