Home
last modified time | relevance | path

Searched refs:scratch_data (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsvdf.h73 int32_t* scratch_data, int32_t* output_temp_data, int32_t scale_1_a, in EvalIntegerSVDF() argument
122 int32_t* scratch_data_batch = scratch_data + b * n_filter; in EvalIntegerSVDF()
132 tensor_utils::ReductionSumVector(scratch_data, output_temp_data, in EvalIntegerSVDF()
157 const float* bias_data, float* scratch_data, float* state_data, in EvalFloatSVDF() argument
173 std::fill_n(scratch_data, batch_size * num_filters, 0.0f); in EvalFloatSVDF()
178 scratch_data); in EvalFloatSVDF()
183 state_data[i * memory_size + memory_size - 1] = scratch_data[i]; in EvalFloatSVDF()
188 bias_data, params->activation, state_data, scratch_data, output_data); in EvalFloatSVDF()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h35 int32_t* col2im_data, int32_t* scratch_data, in TransposeConvV2() argument
76 int32_t* scratch_data_p = scratch_data; in TransposeConvV2()
77 std::fill_n(scratch_data, output_offset * batch_size, static_cast<int32>(0)); in TransposeConvV2()
102 scratch_data_p = scratch_data; in TransposeConvV2()
111 output_min, output_max, scratch_data, output_data); in TransposeConvV2()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h555 int8* scratch_data =
572 memcpy(&scratch_data[0], tmp_load[0][0], 8);
573 memcpy(&scratch_data[8], tmp_load[1][0], 8);
574 memcpy(&scratch_data[16], tmp_load[2][0], 8);
575 memcpy(&scratch_data[24], tmp_load[3][0], 8);
609 int8* scratch_data =
613 memcpy(tmp_load[0][0], &scratch_data[0], 8);
614 memcpy(tmp_load[1][0], &scratch_data[8], 8);
615 memcpy(tmp_load[2][0], &scratch_data[16], 8);
616 memcpy(tmp_load[3][0], &scratch_data[24], 8);
[all …]
Ddepthwiseconv_uint8_3x3_filter.h6833 int8* scratch_data = scratch_data_base + scratch_data_offset;
6846 optimized_ops_prefetch_write_l1_keep(scratch_data);
6847 vst1q_s8(scratch_data, work_reg);
6859 optimized_ops_prefetch_write_l1_keep(scratch_data + start_width +
6861 vst1q_s8(scratch_data + start_width + copy_done, work_reg);
6871 optimized_ops_prefetch_write_l1_keep(scratch_data + start_width +
6873 vst1_s8(scratch_data + start_width + copy_done, half_work_reg);
6902 optimized_ops_prefetch_write_l1_keep(scratch_data + start_width +
6904 vst1_s8(scratch_data + start_width + copy_done, half_work_reg);
6908 optimized_ops_prefetch_write_l1_keep(scratch_data + start_width +
[all …]
Doptimized_ops.h5455 int32_t* col2im_data, int32_t* scratch_data, in TransposeConvV2() argument
5495 int32_t* scratch_data_p = scratch_data; in TransposeConvV2()
5496 std::fill_n(scratch_data, output_offset * batch_size, static_cast<int32>(0)); in TransposeConvV2()
5521 scratch_data_p = scratch_data; in TransposeConvV2()
5526 output_shape.FlatSize(), params.output_offset, scratch_data, in TransposeConvV2()
Dlegacy_optimized_ops.h2557 int32_t* scratch_data, const Dims<4>& scratch_dims, in HybridConv() argument
2574 scratch_data, DimsToShape(output_dims), output_data, in HybridConv()
/external/executorch/backends/arm/runtime/
DArmBackendEthosU.cpp134 handles.scratch_data, in execute()
142 char* scratch_addr = handles.scratch_data + handles.inputs->io[i].offset; in execute()
220 (uint64_t)handles.weight_data, (uint64_t)handles.scratch_data}; in execute()
243 handles.scratch_data + handles.outputs->io[i].offset; in execute()
DVelaBinStream.h57 char* scratch_data; member
DVelaBinStream.cpp75 handles->scratch_data = b->data; in vela_bin_read()
/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/upb/upb/reflection/
Ddef_pool.c50 void* scratch_data; member
57 upb_gfree(s->scratch_data); in upb_DefPool_Free()
69 s->scratch_data = upb_gmalloc(s->scratch_size); in upb_DefPool_New()
70 if (!s->scratch_data) goto err; in upb_DefPool_New()
127 return (void**)&s->scratch_data; in _upb_DefPool_ScratchData()
Dmessage_def.c357 void** scratch_data = _upb_DefPool_ScratchData(ctx->symtab); in _upb_MessageDef_MakeMiniTable() local
360 desc.data, desc.size, ctx->platform, ctx->arena, scratch_data, in _upb_MessageDef_MakeMiniTable()
/external/grpc-grpc/third_party/upb/upb/reflection/
Ddef_pool.c37 void* scratch_data; member
44 upb_gfree(s->scratch_data); in upb_DefPool_Free()
58 s->scratch_data = upb_gmalloc(s->scratch_size); in upb_DefPool_New()
59 if (!s->scratch_data) goto err; in upb_DefPool_New()
181 return (void**)&s->scratch_data; in _upb_DefPool_ScratchData()
Dmessage_def.c359 void** scratch_data = _upb_DefPool_ScratchData(ctx->symtab); in _upb_MessageDef_MakeMiniTable() local
362 desc.data, desc.size, ctx->platform, ctx->arena, scratch_data, in _upb_MessageDef_MakeMiniTable()
/external/flatbuffers/include/flatbuffers/
Dgrpc.h173 uint8_t *buf = buf_.scratch_data(); // pointer to memory
225 auto buf_data = buf_.scratch_data(); // pointer to memory in GetMessage()
Dvector_downward.h183 uint8_t *scratch_data() const { in scratch_data() function
Dflatbuffer_builder.h459 for (auto it = buf_.scratch_data(); it < buf_.scratch_end(); in EndTable()
/external/rust/android-crates-io/crates/ash/src/vk/
Ddefinitions.rs48555 pub scratch_data: DeviceOrHostAddressKHR, field
48595 scratch_data: DeviceOrHostAddressKHR::default(), in default()
48675 pub fn scratch_data(mut self, scratch_data: DeviceOrHostAddressKHR) -> Self { in scratch_data() method
48676 self.inner.scratch_data = scratch_data; in scratch_data()
67826 pub scratch_data: DeviceOrHostAddressKHR, field
67864 scratch_data: DeviceOrHostAddressKHR::default(), in default()
67936 pub fn scratch_data(mut self, scratch_data: DeviceOrHostAddressKHR) -> Self { in scratch_data() method
67937 self.inner.scratch_data = scratch_data; in scratch_data()