| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| cudnn_rnn_ops.py | [all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_benchmark.py | 101 input_h = variables.Variable(
110 input_h=input_h,
115 [params, input_data, input_h, input_c])
|
| cudnn_rnn_ops_test.py | 319 input_h = constant_op.constant(
328 input_h=input_h,
335 input_h=input_h,
435 input_h = array_ops.ones([num_layers * dir_count, batch_size, num_units])
442 input_h=input_h,
449 input_h=input_h,
[all...] |
| cudnn_rnn_test.py | 173 input_h = np.random.randn(num_layers * dir_count, batch_size,
178 initial_state = (input_h, input_c)
180 initial_state = (input_h,)
189 input_h = np.zeros((num_layers * dir_count, batch_size,
194 initial_state = (input_h, input_c)
196 initial_state = (input_h,)
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| cudnn_rnn_ops.cc | 398 const Tensor** input, const Tensor** input_h,
402 TF_RETURN_IF_ERROR(context->input("input_h", input_h));
420 if ((*input_h)->dims() != 3) {
423 model_shapes->num_layers = (*input_h)->dim_size(0) / model_shapes->dir_count;
424 model_shapes->num_units = (*input_h)->dim_size(2);
429 if ((*input_h)->shape() != model_shapes->hidden_state_shape) {
431 "Invalid input_h shape: ", (*input_h)->shape().DebugString(), " ",
435 if ((*input_h)->shape() != (*input_c)->shape())
808 const Tensor* input_h = nullptr; variable
945 const Tensor* input_h = nullptr; variable
[all...] |
| /packages/apps/LegacyCamera/jni/ |
| feature_mos_jni.cpp | 121 void GenerateQuarterResImagePlanar(ImageType im, int input_w, int input_h,
129 for (int j = 0; j < input_h; j += H2L_FACTOR)
141 for (int j = input_h; j < 2 * input_h; j += H2L_FACTOR)
153 for (int j = 2 * input_h; j < 3 * input_h; j += H2L_FACTOR)
|
| /packages/apps/LegacyCamera/jni/feature_stab/src/dbreg/ |
| dbreg.cpp | 746 int input_h = m_im_height*2; local
749 for (int j = 0; j < input_h; j++)
771 for (int j = 2; j < input_h-2; j+=2)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_input_ops.cc | 711 auto input_h = GetTensorDimIndex(data_format, 'H'); local
713 strides[input_h] = row_stride;
715 dilations[input_h] = row_dilation;
[all...] |
