| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 567 const Tensor** input, const Tensor** input_h, in ExtractForwardInput() argument
572 TF_RETURN_IF_ERROR(context->input("input_h", input_h)); in ExtractForwardInput()
595 if ((*input_h)->dims() != 3) { in ExtractForwardInput()
600 (*input_h)->dim_size(0) / model_shapes->dir_count; in ExtractForwardInput()
603 (*input_h)->dim_size(1) / model_shapes->dir_count; in ExtractForwardInput()
605 model_shapes->num_units = (*input_h)->dim_size(2); in ExtractForwardInput()
617 if ((*input_h)->shape() != model_shapes->hidden_state_shape) { in ExtractForwardInput()
619 "Invalid input_h shape: ", (*input_h)->shape().DebugString(), " ", in ExtractForwardInput()
635 if ((*input_h)->shape() != (*input_c)->shape()) { in ExtractForwardInput()
638 (*input_h)->shape().DebugString(), " ", in ExtractForwardInput()
[all …]
|
| D | conv_grad_input_ops.cc | 78 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator ()() local
80 strides[input_h] = row_stride; in operator ()()
82 dilations[input_h] = row_dilation; in operator ()()
|
| D | conv_grad_input_ops.h | 120 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator() local
122 strides[input_h] = row_stride; in operator()
124 dilations[input_h] = row_dilation; in operator()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 18 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
28 output_h: The same shape has input_h.
38 input_h_backprop: The backprop to input_h in the forward pass. Has the same
39 shape as input_h.
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 21 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
31 output_h: The same shape has input_h.
42 input_h_backprop: The backprop to input_h in the forward pass. Has the same
43 shape as input_h.
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 22 input_h: If time_major is true, this is a 3-D tensor with the shape of
35 output_h: The same shape has input_h.
47 input_h_backprop: The backprop to input_h in the forward pass. Has the same
48 shape as input_h.
|
| D | api_def_CudnnRNN.pbtxt | 19 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
29 output_h: The same shape has input_h.
|
| D | api_def_CudnnRNNV2.pbtxt | 20 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
30 output_h: The same shape has input_h.
|
| D | api_def_CudnnRNNV3.pbtxt | 22 input_h: If time_major is true, this is a 3-D tensor with the shape of
35 output_h: The same shape has input_h.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cudnn_rnn_grad.py | 32 input_h=op.inputs[1],
57 input_h=op.inputs[1],
85 input_h=op.inputs[1],
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | cudnn_recurrent.py | 275 input_h = initial_state[0]
276 input_h = array_ops.expand_dims(input_h, axis=0)
299 'input_h': input_h,
471 input_h = initial_state[0]
473 input_h = array_ops.expand_dims(input_h, axis=0)
501 'input_h': input_h,
|
| D | recurrent_v2.py | 677 input_h=init_h,
693 input=inputs, input_h=init_h, input_c=0, params=params,
1494 input_h=init_h,
1513 input=inputs, input_h=init_h, input_c=init_c, params=params,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CudnnRNNV3.pbtxt | 8 name: "input_h"
133 name: "input_h"
265 name: "input_h"
|
| D | CudnnRNNBackpropV3.pbtxt | 8 name: "input_h"
154 name: "input_h"
307 name: "input_h"
|
| D | CudnnRNN.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNV2.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNBackpropV2.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNBackprop.pbtxt | 8 name: "input_h"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CudnnRNNV3.pbtxt | 8 name: "input_h"
133 name: "input_h"
|
| D | CudnnRNNBackpropV3.pbtxt | 8 name: "input_h"
154 name: "input_h"
|
| D | CudnnRNN.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNV2.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNBackpropV2.pbtxt | 8 name: "input_h"
|
| D | CudnnRNNBackprop.pbtxt | 8 name: "input_h"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 945 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
949 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'outpu…
953 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'output\', \'output_h\', \'outpu…
957 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'output\',…
981 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'rnn_mode\', \'input_mode\', \'d…
985 …argspec: "args=[\'input\', \'input_h\', \'input_c\', \'params\', \'sequence_lengths\', \'rnn_mode\…
|