| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | gru_ops.cc | 54 const int64 cell_size = h_prev_tensor->dim_size(1); in Compute() local
63 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size, in Compute()
66 " vs. ", cell_size)); in Compute()
69 OP_REQUIRES(ctx, w_ru_tensor->dim_size(0) == input_size + cell_size, in Compute()
72 w_ru_tensor->dim_size(0), " vs. ", input_size + cell_size)); in Compute()
74 OP_REQUIRES(ctx, w_ru_tensor->dim_size(1) == cell_size * 2, in Compute()
77 cell_size * 2)); in Compute()
80 OP_REQUIRES(ctx, w_c_tensor->dim_size(0) == input_size + cell_size, in Compute()
83 w_c_tensor->dim_size(0), " vs. ", input_size + cell_size)); in Compute()
85 OP_REQUIRES(ctx, w_c_tensor->dim_size(1) == cell_size, in Compute()
[all …]
|
| D | lstm_ops_gpu.cu.cc | 90 const int batch_size, const int cell_size) { in lstm_gates() argument
97 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates()
142 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates()
143 const int cid = batch_id * cell_size + act_id; in lstm_gates()
150 i_local = sigmoid_op(icfo[0 * cell_size + gid] + b[0 * cell_size + act_id] + in lstm_gates()
153 i_local = sigmoid_op(icfo[0 * cell_size + gid] + b[0 * cell_size + act_id]); in lstm_gates()
158 tanh_op(icfo[1 * cell_size + gid] + b[1 * cell_size + act_id]); in lstm_gates()
163 f_local = sigmoid_op(icfo[2 * cell_size + gid] + b[2 * cell_size + act_id] + in lstm_gates()
166 f_local = sigmoid_op(icfo[2 * cell_size + gid] + b[2 * cell_size + act_id] + in lstm_gates()
182 o_local = sigmoid_op(icfo[3 * cell_size + gid] + b[3 * cell_size + act_id] + in lstm_gates()
[all …]
|
| D | lstm_ops.cc | 70 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellFpropWithEigen()
149 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellBpropWithEigen()
270 const int64 cell_size = cs_prev_tensor->dim_size(1); in Compute() local
277 OP_REQUIRES(ctx, cs_prev_tensor->dim_size(1) == cell_size, in Compute()
280 cell_size)); in Compute()
286 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size, in Compute()
289 " vs. ", cell_size)); in Compute()
291 OP_REQUIRES(ctx, w_tensor->dim_size(0) == input_size + cell_size, in Compute()
294 w_tensor->dim_size(0), " vs. ", input_size + cell_size)); in Compute()
295 OP_REQUIRES(ctx, w_tensor->dim_size(1) == cell_size * 4, in Compute()
[all …]
|
| D | lstm_ops.h | 95 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size) in LSTMBlockCell()
98 cell_size_(cell_size) {} in LSTMBlockCell()
104 int cell_size() const { return cell_size_; } in cell_size() function
153 const int cell_size) in LSTMBlockCellFprop()
154 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellFprop()
178 const int cell_size) in LSTMBlockCellBprop()
179 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellBprop()
203 const int cell_size) in BlockLSTMBprop()
204 : LSTMBlockCell(batch_size, input_size, cell_size) {} in BlockLSTMBprop()
|
| D | gru_ops.h | 31 GRUCell(const int batch_size, const int input_size, const int cell_size) in GRUCell()
34 cell_size_(cell_size) {} in GRUCell()
71 const int cell_size) in GRUBlockCellFprop()
72 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellFprop()
127 const int cell_size) in GRUBlockCellBprop()
128 : GRUCell(batch_size, input_size, cell_size) {} in GRUBlockCellBprop()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | gru_ops_test.py | 46 cell_size = 5
50 cell = gru_ops.GRUBlockCell(cell_size)
63 cell_size = 5
71 h = array_ops.zeros([batch_size, cell_size])
75 h_value = np.random.rand(batch_size, cell_size)
79 output = rnn_cell.GRUCell(cell_size)(x, h)
85 output = gru_ops.GRUBlockCell(cell_size)(x, h)
96 cell_size = 3
108 h = array_ops.zeros([batch_size, cell_size])
112 h_value = np.random.rand(batch_size, cell_size)
[all …]
|
| D | lstm_ops_test.py | 110 cell_size = 4
127 "rnn/lstm_cell/w_i_diag", shape=[cell_size], dtype=dtype)
129 "rnn/lstm_cell/w_f_diag", shape=[cell_size], dtype=dtype)
131 "rnn/lstm_cell/w_o_diag", shape=[cell_size], dtype=dtype)
135 shape=[input_size + cell_size, cell_size * 4],
139 shape=[cell_size * 4],
144 cell_size,
173 cell_size,
223 cell_size = 7
225 cell = lstm_ops.LSTMBlockCell(cell_size)
[all …]
|
| /external/tensorflow/tensorflow/contrib/rnn/ops/ |
| D | lstm_ops.cc | 51 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010102() local
52 ShapeHandle output = c->Matrix(batch_size, cell_size); in __anon4050c6010102()
138 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4050c6010202() local
140 TF_RETURN_IF_ERROR(c->Multiply(cell_size, 4, &cell_size_times_4)); in __anon4050c6010202()
141 ShapeHandle cell_size_vec = c->Vector(cell_size); in __anon4050c6010202()
143 c->set_output(0, c->Matrix(batch_size, cell_size)); in __anon4050c6010202()
207 DimensionHandle cell_size; in __anon4050c6010302() local
209 c->Divide(c->Dim(b, 0), 4, true /* evenly_divisible */, &cell_size)); in __anon4050c6010302()
212 ShapeHandle output = c->MakeShape({timelen, batch_size, cell_size}); in __anon4050c6010302()
|
| D | gru_ops.cc | 41 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon07a5ee8f0102() local
42 ShapeHandle output = c->Matrix(batch_size, cell_size); in __anon07a5ee8f0102()
120 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon07a5ee8f0202() local
122 ShapeHandle batch_cell_shape = c->Matrix(batch_size, cell_size); in __anon07a5ee8f0202()
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | gru_ops.py | 140 cell_size=None, argument
161 if (cell_size is None) == (num_units is None):
165 num_units = cell_size
200 cell_size = h_prev.get_shape().with_rank(2)[1]
201 if cell_size != self._cell_size:
203 (self._cell_size, cell_size))
|
| D | lstm_ops.py | 119 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value
120 if cell_size is None:
122 wci = array_ops.constant(0, dtype=dtypes.float32, shape=[cell_size])
198 cell_size = cell_size4 / 4
202 0, dtype=dtype, shape=[batch_size, cell_size])
208 wci = array_ops.constant(0, dtype=dtype, shape=[cell_size])
251 cell_size = cs_prev.get_shape().with_rank(2).dims[1].value
252 if cell_size is None:
282 (batch_size, cell_size))
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dynamic_rnn_estimator_test.py | 319 array_ops.expand_dims(math_ops.range(cell_size), 0)
320 for i, cell_size in enumerate([5, 5, 3, 3, 7, 7])
381 [batch_size, cell_size], seed=((i + 1) * seed))
382 for i, cell_size in enumerate([4, 4, 8, 8, 7, 7])
514 cell_size = [4]
542 'inputs', dimension=cell_size[0])
548 num_units=cell_size,
576 cell_size = 4
603 'inputs', dimension=cell_size)
610 num_units=cell_size,
[all …]
|
| /external/iproute2/tc/ |
| D | q_netem.c | 415 if (get_u32(&rate.cell_size, *argv, 0)) { in netem_parse_opt()
668 if (rate->cell_size) in netem_print_opt()
669 fprintf(f, " cellsize %u", rate->cell_size); in netem_print_opt()
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/kernel_tests/ |
| D | functional_rnn_test.py | 40 subcells = [rnn_cell_impl.LSTMCell(cell_size) for cell_size in cell_sizes]
|
| /external/libnl/lib/route/ |
| D | tc.c | 617 int rtnl_tc_calc_cell_log(int cell_size) in rtnl_tc_calc_cell_log() argument
622 if ((1 << i) == cell_size) in rtnl_tc_calc_cell_log()
|
| /external/iproute2/include/uapi/linux/ |
| D | pkt_sched.h | 571 __u32 cell_size; member
|
| /external/kernel-headers/original/uapi/linux/ |
| D | pkt_sched.h | 618 __u32 cell_size; member
|
| /external/syzkaller/sys/linux/ |
| D | socket_netlink_route_sched.txt | 390 cell_size int32
|