Home
last modified time | relevance | path

Searched defs:num_elems (Results 1 – 25 of 54) sorted by relevance

123

/external/rappor/client/cpp/
Dunix_kernel_rand_impl.cc25 size_t num_elems = fread(&rand_buf, sizeof(uint8_t), num_bits, fp_); in GetMask() local
/external/tensorflow/tensorflow/core/kernels/
Dsparse_to_dense_op.cc61 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
Dsparse_to_dense_op_gpu.cu.cc113 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
Dsparse_reorder_op_gpu.cu.cc74 const int64 num_elems = input_ind.dims() > 0 ? input_ind.dim_size(0) : 1; in operator ()() local
Dtensor_to_hash_bucket_op_gpu.cu.cc89 const int num_elems, int64* output) { in operator ()()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.cc104 int64_t num_elems = 1; in ComputeBytesFromShape() local
/external/libnetfilter_conntrack/src/conntrack/
Dstack.c18 int num_elems; member
/external/rust/crates/quiche/deps/boringssl/src/crypto/test/
Dmalloc.cc123 void *calloc(size_t num_elems, size_t size) { in calloc()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsparse_to_dense_op.cc37 const int64_t num_elems = in Compile() local
/external/sg3_utils/testing/
Dsg_scat_gath.cpp52 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
/external/pdfium/core/fxcrt/
Dfx_memory.cpp25 void* FXMEM_DefaultCalloc(size_t num_elems, size_t byte_size) { in FXMEM_DefaultCalloc()
/external/tensorflow/tensorflow/lite/experimental/remat/
Dmetadata_util.cc89 uint32_t num_elems = 0; in Parse() local
/external/mesa3d/src/compiler/glsl/
Dir_set_program_inouts.cpp283 unsigned num_elems; in try_mark_partial_variable() local
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_expander.cc40 int64_t num_elems = ShapeUtil::ElementsIn(shape); in GetNumberOf32bitUnits() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/
Dconversion.cc68 const int64_t num_elems = tf_tensor->NumElements(); in ConvertKernelFallbackTensorToStringHostTensor() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_tensor.cc86 const int64_t num_elems = tensor_interface->NumElements(); in CopyTfStringTensorToStringHostTensor() local
/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc109 int num_elems = TF_TensorElementCount(theta_tensor); in CalcNumericalGrad() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc391 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
/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py1427 def cond(i, num_elems, *args): argument
1433 def body(i, num_elems, *args): argument
/external/parameter-framework/asio-1.10.6/include/asio/detail/
Dhash_map.hpp210 static std::size_t hash_size(std::size_t num_elems) in hash_size()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc159 string MakeFloatTensorTestCase(int num_elems) { in MakeFloatTensorTestCase()
/external/flatbuffers/include/flatbuffers/
Dflatbuffers.h220 size_t num_elems; // of type_codes, values, names (but not type_refs). member
/external/linux-kselftest/tools/testing/selftests/bpf/prog_tests/
Dsockmap_basic.c204 __u32 i, num_sockets, num_elems; in test_sockmap_copy() local
/external/mesa3d/src/compiler/nir/
Dnir_gather_info.c268 unsigned num_elems; in try_mask_partial_io() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc115 int num_elems = TensorShape(shape).num_elements(); in InferMemUsageForNeighbors() local

123