Searched refs:params_size (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNParamsSize.pbtxt | 21 params_size: The size of the params buffer that should be allocated and
|
| /external/kernel-headers/original/uapi/drm/ |
| D | exynos_drm.h | 346 __u32 params_size; member
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_benchmark.py | 98 params_size_t = model.params_size()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_op_test.py | 411 params_size = np.prod(params_shape)
412 params = np.reshape(np.arange(params_size), params_shape)
|
| D | resource_variable_ops_test.py | 1209 params_size = np.prod(params_shape)
1210 params = np.reshape(np.arange(params_size, dtype=np.int32), params_shape)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1046 int64 params_size = params_size_in_bytes / sizeof(T); in Compute() local
1050 *output_t->template flat<Index>().data() = params_size; in Compute()
1219 int params_size = params_size_in_bytes / sizeof(T); in Compute() local
1221 context->allocate_output(0, {params_size}, &output)); in Compute()
|
| /external/webp/src/enc/ |
| D | vp8l_enc.c | 1723 const int params_size = (num_crunch_configs_side > 0) ? 2 : 1; in VP8LEncodeStream() local
1724 for (idx = 0; idx < params_size; ++idx) { in VP8LEncodeStream()
|
| /external/nos/host/android/hals/keymaster/ |
| D | proto_utils.cpp | 837 for (size_t i = 0; i < (size_t)pbParams.params_size(); i++) { in pb_to_hidl_params()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 1573 def params_size(self): member in _CudnnRNN
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1170 size_t params_size = 0; in Create() local
1173 /*xDesc=*/input_desc.get(), /*sizeInBytes=*/¶ms_size, in Create()
1175 int64 params_size_in_bytes = static_cast<int64>(params_size); in Create()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2247 size_t params_size = 0; in MIOpenRnnParamsDescriptor() local
2250 input_desc /*xDesc*/, ¶ms_size /*sizeInBytes*/, in MIOpenRnnParamsDescriptor()
2253 params_size_in_bytes_ = static_cast<int64>(params_size); in MIOpenRnnParamsDescriptor()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 32802 …Output, T tf.DataType, S tf.DataType, optional ...CudnnRNNParamsSizeAttr) (params_size tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7896 name: "params_size"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 17293 name: "params_size"
|
| D | ops_history.v1.pbtxt | 17504 name: "params_size"
|