Home
last modified time | relevance | path

Searched refs:memory_size (Results 1 – 25 of 27) sorted by relevance

12

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DSVDF.cpp85 const uint32_t memory_size = SizeOfDimension(weights_time, 1); in Prepare() local
97 stateShape->dimensions = {batch_size, memory_size * num_filters}; in Prepare()
172 const int memory_size = SizeOfDimension(weights_time_, 1); in EvalFloat32() local
177 float* state_ptr_batch = outputStateData + b * memory_size * num_filters; in EvalFloat32()
179 float* state_ptr = state_ptr_batch + c * memory_size; in EvalFloat32()
180 state_ptr[memory_size - 1] = 0.0; in EvalFloat32()
193 outputStateData[i * memory_size + memory_size - 1] = scratch[i]; in EvalFloat32()
199 float* state_out_ptr_batch = outputStateData + b * memory_size * num_filters; in EvalFloat32()
222 std::copy(state_out_ptr_batch + 1, state_out_ptr_batch + memory_size, in EvalFloat32()
224 state_out_ptr_batch[memory_size - 1] = 0.0; in EvalFloat32()
[all …]
H A DSVDFTest.cpp168 SVDFOpModel(uint32_t batches, uint32_t units, uint32_t input_size, uint32_t memory_size, in SVDFOpModel() argument
173 memory_size_(memory_size), in SVDFOpModel()
180 {batches_, memory_size * units_ * rank_}, // state in tensor in SVDFOpModel()
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
H A Dsvdf.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
60 state_in: [0 for _ in range(batches * memory_size * features)],
127 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf2.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
75 state_in: [0 for _ in range(batches * memory_size * features)],
142 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_bias_present.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
60 state_in: [0 for _ in range(batches * memory_size * features)],
127 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_state.mod.py20 memory_size = 10 variable
26 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (units, memory_size))
28 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units))
31 state_out = Output("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units))
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
H A Dsvdf_bias_present_float16.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT16", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*feature…
60 state_in: [0 for _ in range(batches * memory_size * features)],
127 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_float16.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT16", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*feature…
60 state_in: [0 for _ in range(batches * memory_size * features)],
127 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_state_float16.mod.py20 memory_size = 10 variable
26 weights_time = Input("weights_time", "TENSOR_FLOAT16", "{%d, %d}" % (units, memory_size))
28 state_in = Input("state_in", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*units))
31 state_out = Output("state_out", "TENSOR_FLOAT16", "{%d, %d}" % (batches, memory_size*units))
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
H A Dsvdf2_relaxed.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
76 state_in: [0 for _ in range(batches * memory_size * features)],
143 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_bias_present_relaxed.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
61 state_in: [0 for _ in range(batches * memory_size * features)],
128 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_relaxed.mod.py22 memory_size = 10 variable
28 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (features, memory_size))
30 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*features))
33 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*feature…
61 state_in: [0 for _ in range(batches * memory_size * features)],
128 output0 = {state_out: [0 for _ in range(batches * memory_size * features)],
H A Dsvdf_state_relaxed.mod.py20 memory_size = 10 variable
26 weights_time = Input("weights_time", "TENSOR_FLOAT32", "{%d, %d}" % (units, memory_size))
28 state_in = Input("state_in", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units))
31 state_out = Output("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units))
/aosp12/system/core/init/
H A Dservice_test.cpp33 constexpr auto memory_size = sizeof(Service); in TEST() local
34 alignas(alignof(Service)) unsigned char old_memory[memory_size]; in TEST()
36 for (std::size_t i = 0; i < memory_size; ++i) { in TEST()
56 for (std::size_t i = 0; i < memory_size; ++i) { in TEST()
/aosp12/frameworks/native/libs/vr/libbroadcastring/include/libbroadcastring/
H A Dbroadcast_ring.h499 size_t memory_size = record_count() * record_size(); in ValidateGeometry() local
500 if (memory_size / record_size() != record_count()) return false; in ValidateGeometry()
501 if (memory_size + sizeof(Header) < memory_size) return false; in ValidateGeometry()
502 if (memory_size + sizeof(Header) > mmap_size) return false; in ValidateGeometry()
/aosp12/system/bt/test/mock/
H A Dmock_btif_co_bta_dm_co.cc36 UNUSED_ATTR uint32_t* memory_size) { in bta_dm_co_get_compress_memory() argument
/aosp12/system/bt/btif/co/
H A Dbta_dm_co.cc56 UNUSED_ATTR uint32_t* memory_size) { in bta_dm_co_get_compress_memory() argument
/aosp12/bionic/libc/kernel/uapi/linux/
H A Dnitro_enclaves.h57 __u64 memory_size; member
H A Dvhost_types.h77 __u64 memory_size; member
H A Dkvm.h83 __u64 memory_size; member
89 __u64 memory_size; member
/aosp12/frameworks/minikin/tests/util/
H A DFreeTypeMinikinFontForTest.cpp78 args.memory_size = mFontSize; in FreeTypeMinikinFontForTest()
/aosp12/art/libdexfile/dex/
H A Ddex_file_loader.cc231 auto memory_size = memory.size(); in Open() local
233 memory_size, in Open()
/aosp12/bionic/libc/kernel/uapi/asm-x86/asm/
H A Dkvm.h64 __u64 memory_size; member
/aosp12/hardware/interfaces/neuralnetworks/1.0/
H A Dtypes.hal1373 * get pushed into a memory of fixed-size memory_size.
1374 * * stage 2 performs filtering on the "time" dimension of the memory_size
1412 * A 2-D tensor of shape [num_units, memory_size], where “memory_size”
1417 * A 2-D tensor of shape [batch_size, (memory_size - 1) * num_units * rank].
1428 * [batch_size, (memory_size - 1) * num_units * rank].
/aosp12/packages/modules/NeuralNetworks/tools/api/
H A Dtypes.spec2365 * get pushed into a memory of fixed-size memory_size.
2366 * * stage 2 performs filtering on the "time" dimension of the memory_size
2407 * A 2-D tensor of shape [num_units, memory_size], where “memory_size”
2412 * A 2-D tensor of shape [batch_size, (memory_size - 1) * num_units * rank].
2423 * [batch_size, (memory_size - 1) * num_units * rank].

12