Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.cc104 int64_t num_elems = 1; in ComputeBytesFromShape() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsparse_to_dense_op.cc37 const int64 num_elems = in Compile() local
/external/libnetfilter_conntrack/src/conntrack/
Dstack.c18 int num_elems; member
/external/boringssl/src/crypto/test/
Dmalloc.cc123 void *calloc(size_t num_elems, size_t size) { in calloc()
/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/core/kernels/
Dsparse_to_dense_op.cc59 const int64 num_elems = indices.dims() > 0 ? indices.dim_size(0) : 1; in Compute() local
/external/pdfium/core/fxcrt/
Dfx_memory.cpp49 void* FXMEM_DefaultCalloc(size_t num_elems, size_t byte_size) { in FXMEM_DefaultCalloc()
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_expander.cc38 int64 num_elems = ShapeUtil::ElementsIn(shape); in GetNumberOf32bitUnits() local
/external/mesa3d/src/compiler/glsl/
Dir_set_program_inouts.cpp283 unsigned num_elems; in try_mark_partial_variable() local
/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc108 int num_elems = TF_TensorElementCount(theta_tensor); in CalcNumericalGrad() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc390 std::pair<Philox4x32State, XlaOp> GeneratePhiloxBits(int64 num_elems, in GeneratePhiloxBits()
410 const int64 num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit32() local
437 const int64 num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit64() local
/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/python/ops/
Dctc_ops.py1369 def cond(i, num_elems, *args): argument
1375 def body(i, num_elems, *args): argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc159 string MakeFloatTensorTestCase(int num_elems) { in MakeFloatTensorTestCase()
/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
Dutils.cc219 int64 num_elems = TensorShape(shape).num_elements(); in CalculateTensorSize() local
/external/flatbuffers/java/com/google/flatbuffers/
DFlatBufferBuilder.java459 public void startVector(int elem_size, int num_elems, int alignment) { in startVector()
493 public ByteBuffer createUnintializedVector(int elem_size, int num_elems, int alignment) { in createUnintializedVector()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Ddesc_sroa.cpp373 uint32_t num_elems = length_const->GetU32(); in GetNumBindingsUsedByType() local
/external/deqp-deps/SPIRV-Tools/source/opt/
Ddesc_sroa.cpp373 uint32_t num_elems = length_const->GetU32(); in GetNumBindingsUsedByType() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Ddesc_sroa.cpp373 uint32_t num_elems = length_const->GetU32(); in GetNumBindingsUsedByType() local
/external/llvm-project/lldb/source/Breakpoint/
DBreakpointOptions.cpp95 size_t num_elems = user_source->GetSize(); in CreateFromStructuredData() local
/external/flatbuffers/php/
DFlatbufferBuilder.php584 public function startVector($elem_size, $num_elems, $alignment)
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.cc45 int64 num_elems = TensorShape(t.tensor_shape()).num_elements(); in TensorByteSize() local
Dtensor.cc877 const int64 num_elems = dim0_size * elems_per_dim0; in Slice() local
899 const int64 num_elems = elems_per_dim0; in SubSlice() local

12