| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| 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 | 89 const int batch_size, const int cell_size) { in lstm_gates() argument
96 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()
151 sigmoid_op(gates[0 * cell_size + gid] + b[0 * cell_size + act_id] + in lstm_gates()
155 sigmoid_op(gates[0 * cell_size + gid] + b[0 * cell_size + act_id]); in lstm_gates()
159 const int c_offset = gate_c_offset(gate_layout, cell_size); in lstm_gates()
160 const int f_offset = gate_f_offset(gate_layout, cell_size); in lstm_gates()
186 o_local = sigmoid_op(gates[3 * cell_size + gid] + in lstm_gates()
187 b[3 * cell_size + act_id] + cs_local * wco[act_id]); in lstm_gates()
[all …]
|
| D | lstm_ops.cc | 70 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellFpropWithEigen()
153 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellBpropWithEigen()
333 const int64 cell_size = cs_prev_tensor->dim_size(1); in Compute() local
340 OP_REQUIRES(ctx, cs_prev_tensor->dim_size(1) == cell_size, in Compute()
343 cell_size)); in Compute()
349 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size, in Compute()
352 " vs. ", cell_size)); in Compute()
354 OP_REQUIRES(ctx, w_tensor->dim_size(0) == input_size + cell_size, in Compute()
357 w_tensor->dim_size(0), " vs. ", input_size + cell_size)); in Compute()
358 OP_REQUIRES(ctx, w_tensor->dim_size(1) == cell_size * 4, in Compute()
[all …]
|
| D | lstm_ops.h | 30 constexpr int gate_c_offset(GateLayout gate_layout, int cell_size) { in gate_c_offset() argument
31 return (gate_layout == ICFO) ? cell_size : cell_size * 2; in gate_c_offset()
34 constexpr int gate_f_offset(GateLayout gate_layout, int cell_size) { in gate_f_offset() argument
35 return (gate_layout == ICFO) ? cell_size * 2 : cell_size; in gate_f_offset()
105 LSTMBlockCell(const int batch_size, const int input_size, const int cell_size) in LSTMBlockCell()
108 cell_size_(cell_size) {} in LSTMBlockCell()
114 int cell_size() const { return cell_size_; } in cell_size() function
165 const int cell_size) in LSTMBlockCellFprop()
166 : LSTMBlockCell(batch_size, input_size, cell_size) {} in LSTMBlockCellFprop()
190 const int cell_size) in LSTMBlockCellBprop()
[all …]
|
| 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/core/ops/ |
| D | rnn_ops.cc | 43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880102() local
44 ShapeHandle output = c->Matrix(batch_size, cell_size); in __anon652ace880102()
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon652ace880202() local
76 ShapeHandle batch_cell_shape = c->Matrix(batch_size, cell_size); in __anon652ace880202()
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880302() local
112 ShapeHandle output = c->Matrix(batch_size, cell_size); in __anon652ace880302()
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon652ace880402() local
151 TF_RETURN_IF_ERROR(c->Multiply(cell_size, 4, &cell_size_times_4)); in __anon652ace880402()
152 ShapeHandle cell_size_vec = c->Vector(cell_size); in __anon652ace880402()
154 c->set_output(0, c->Matrix(batch_size, cell_size)); in __anon652ace880402()
[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/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1060 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor() argument
1073 cell_size_(cell_size), in CudnnRnnDescriptor()
1090 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create() argument
1104 hidden_size = std::max(hidden_size, cell_size); in Create()
1195 num_layers, hidden_size, input_size, cell_size, in Create()
1205 int cell_size() const { return cell_size_; } in cell_size() function in stream_executor::gpu::CudnnRnnDescriptor
1592 int cell_size = 0; member
1618 model_dims.cell_size = rnn_desc.cell_size(); in ExtractAndCheckRnnForward()
2071 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
2083 cudnn, num_layers, hidden_size, input_size, cell_size, batch_size, in createRnnDescriptor()
|
| D | cuda_dnn.h | 51 int num_layers, int hidden_size, int input_size, int cell_size,
|
| /external/libnl/lib/route/ |
| D | tc.c | 678 int rtnl_tc_calc_cell_log(int cell_size) in rtnl_tc_calc_cell_log() argument
683 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/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 366 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
378 num_layers, hidden_size, input_size, cell_size, batch_size, input_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 416 int num_layers, int hidden_size, int input_size, int cell_size,
|
| D | dnn.h | 2111 int cell_size, int batch_size, in createRnnDescriptor() argument
|
| /external/libnl/include/linux-private/linux/ |
| D | pkt_sched.h | 591 __u32 cell_size; member
|
| /external/kernel-headers/original/uapi/linux/ |
| D | pkt_sched.h | 639 __u32 cell_size; member
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 85 int num_layers, int hidden_size, int input_size, int cell_size,
|
| D | rocm_dnn.cc | 2516 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor() argument
2529 bool use_projection = cell_size != 0 && hidden_size < cell_size; in createRnnDescriptor()
|