Home
last modified time | relevance | path

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

12

/external/pthreadpool/src/
Dgcd.c80 size_t params_size, in pthreadpool_parallelize() argument
103 if (params_size != 0) { in pthreadpool_parallelize()
104 memcpy(&threadpool->params, params, params_size); in pthreadpool_parallelize()
Dwindows.c194 size_t params_size, in pthreadpool_parallelize() argument
218 if (params_size != 0) { in pthreadpool_parallelize()
219 CopyMemory(&threadpool->params, params, params_size); in pthreadpool_parallelize()
Dpthreads.c298 size_t params_size, in pthreadpool_parallelize() argument
330 if (params_size != 0) { in pthreadpool_parallelize()
331 memcpy(&threadpool->params, params, params_size); in pthreadpool_parallelize()
Dthreadpool-object.h724 size_t params_size,
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dimport_quant_stats_pass.cc138 if (info.params_size() == 0) return; in ImportAsStatsOps()
141 min_maxs.reserve(info.params_size() * 2); in ImportAsStatsOps()
154 if (info.params_size() > 1) { in ImportAsStatsOps()
155 SmallVector<int64_t, 4> axis_stats_shape{info.params_size(), 2}; in ImportAsStatsOps()
/external/XNNPACK/src/operators/
Dunary-elementwise-nc.c26 size_t params_size, in create_unary_elementwise_nc() argument
73 if (params_size != 0) { in create_unary_elementwise_nc()
74 memcpy(&unary_elementwise_op->params, params, params_size); in create_unary_elementwise_nc()
93 size_t params_size) in setup_unary_elementwise_nc() argument
121 if (params_size != 0) { in setup_unary_elementwise_nc()
122 memcpy(&unary_elementwise_op->context.univector_contiguous.params, params, params_size); in setup_unary_elementwise_nc()
137 if (params_size != 0) { in setup_unary_elementwise_nc()
138 memcpy(&unary_elementwise_op->context.univector_strided.params, params, params_size); in setup_unary_elementwise_nc()
Ddeconvolution-nhwc.c69 size_t params_size, in create_deconvolution2d_nhwc()
262 memcpy(&deconvolution_op->params, params, params_size); in create_deconvolution2d_nhwc()
486 size_t params_size, in setup_conv_path() argument
547 memcpy(&deconvolution_op->context.igemm.params, params, params_size); in setup_conv_path()
609 size_t params_size, in setup_subconv2d_path() argument
702 memcpy(&deconvolution_op->context.subgemm.params, params, params_size); in setup_subconv2d_path()
720 memcpy(&deconvolution_op->context.subconv.params, params, params_size); in setup_subconv2d_path()
780 size_t params_size, in setup_deconvolution2d_nhwc() argument
854 params, params_size, num_threads); in setup_deconvolution2d_nhwc()
869 params, params_size, num_threads, use_gemm); in setup_deconvolution2d_nhwc()
Dfully-connected-nc.c41 size_t params_size, in create_fully_connected_nc() argument
136 memcpy(&fully_connected_op->params, params, params_size); in create_fully_connected_nc()
168 size_t params_size, in setup_fully_connected_nc() argument
217 memcpy(&fully_connected_op->context.gemm.params, params, params_size); in setup_fully_connected_nc()
Dbinary-elementwise-nd.c25 size_t params_size, in create_binary_elementwise_nd() argument
51 if (params_size != 0) { in create_binary_elementwise_nd()
52 memcpy(&binary_elementwise_op->params, params, params_size); in create_binary_elementwise_nd()
385 size_t params_size, in setup_binary_elementwise_nd() argument
521 if (params_size != 0) { in setup_binary_elementwise_nd()
522 memcpy(&binary_elementwise_op->context.elementwise_binary.params, params, params_size); in setup_binary_elementwise_nd()
Dmax-pooling-nhwc.c61 size_t params_size, in create_max_pooling2d_nhwc() argument
177 memcpy(&max_pooling_op->params, params, params_size); in create_max_pooling2d_nhwc()
202 size_t params_size, in setup_max_pooling2d_nhwc() argument
313 memcpy(&max_pooling_op->context.max_pooling.params, params, params_size); in setup_max_pooling2d_nhwc()
Dglobal-average-pooling-nwc.c33 size_t params_size, in create_global_average_pooling_nwc() argument
103 memcpy((void*) ((uintptr_t) global_average_pooling_op + params_offset), params, params_size); in create_global_average_pooling_nwc()
128 size_t params_size, in setup_global_average_pooling_nwc() argument
184 …memcpy(&global_average_pooling_op->context.global_average_pooling_nwc.params, params, params_size); in setup_global_average_pooling_nwc()
Daverage-pooling-nhwc.c415 size_t params_size, in setup_average_pooling2d() argument
594 memcpy(&average_pooling_op->context.pixelwise_average_pooling.params, params, params_size); in setup_average_pooling2d()
622 memcpy(&average_pooling_op->context.average_pooling.params, params, params_size); in setup_average_pooling2d()
Dconvolution-nhwc.c93 size_t params_size, in create_convolution2d_nhwc()
401 memcpy(&convolution_op->params, params, params_size); in create_convolution2d_nhwc()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DCudnnRNNParamsSize.pbtxt16 name: "params_size"
119 name: "params_size"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DCudnnRNNParamsSize.pbtxt16 name: "params_size"
119 name: "params_size"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CudnnRNNParamsSize.pbtxt21 params_size: The size of the params buffer that should be allocated and
/external/kernel-headers/original/uapi/drm/
Dexynos_drm.h346 __u32 params_size; member
/external/igt-gpu-tools/include/drm-uapi/
Dexynos_drm.h346 __u32 params_size; member
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dgather_op_test.py542 params_size = np.prod(params_shape)
543 params = np.reshape(np.arange(params_size), params_shape)
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1133 int64 params_size = params_size_in_bytes / sizeof(T); in Compute() local
1137 *output_t->template flat<Index>().data() = params_size; in Compute()
1384 int params_size = params_size_in_bytes / sizeof(T); in Compute() local
1386 context->allocate_output(0, {params_size}, &output)); in Compute()
/external/webp/src/enc/
Dvp8l_enc.c1736 const int params_size = (num_crunch_configs_side > 0) ? 2 : 1; in VP8LEncodeStream() local
1737 for (idx = 0; idx < params_size; ++idx) { in VP8LEncodeStream()
/external/tensorflow/tensorflow/python/kernel_tests/
Dresource_variable_ops_test.py1532 params_size = np.prod(params_shape)
1533 params = np.reshape(np.arange(params_size, dtype=np.int32), params_shape)
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1369 size_t params_size = 0; in Create() local
1372 /*xDesc=*/input_desc.get(), /*sizeInBytes=*/&params_size, in Create()
1374 int64 params_size_in_bytes = static_cast<int64>(params_size); in Create()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2329 size_t params_size = 0; in MIOpenRnnParamsDescriptor() local
2332 input_desc /*xDesc*/, &params_size /*sizeInBytes*/, in MIOpenRnnParamsDescriptor()
2335 params_size_in_bytes_ = static_cast<int64>(params_size); in MIOpenRnnParamsDescriptor()
/external/tensorflow/tensorflow/go/op/
Dwrappers.go5999 …Output, T tf.DataType, S tf.DataType, optional ...CudnnRNNParamsSizeAttr) (params_size tf.Output) {

12