| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops.cc | 54 const int64 cell_size = h_prev_tensor->dim_size(1); variable
58 // Shape of 'h' must be [batch_size, cell_size]
63 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size,
65 "h_prev.dims(1) != cell_size: ", h_prev_tensor->dim_size(1),
66 " vs. ", cell_size));
68 // Shape of 'w_ru' must be [input_size+cell_size, 2*cell_size]
69 OP_REQUIRES(ctx, w_ru_tensor->dim_size(0) == input_size + cell_size,
71 "w_ru.dim_size(0) != input_size + cell_size: ",
72 w_ru_tensor->dim_size(0), " vs. ", input_size + cell_size));
209 const int64 cell_size = h_prev_tensor->dim_size(1); variable
[all...] |
| lstm_ops.cc | 69 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()});
148 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()});
267 const int64 cell_size = cs_prev_tensor->dim_size(1); variable
274 OP_REQUIRES(ctx, cs_prev_tensor->dim_size(1) == cell_size,
275 errors::InvalidArgument("cs_prev.dims(1) != cell_size: ",
277 cell_size));
283 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size,
285 "h_prev.dims(1) != cell_size: ", h_prev_tensor->dim_size(1),
286 " vs. ", cell_size));
288 OP_REQUIRES(ctx, w_tensor->dim_size(0) == input_size + cell_size,
470 const int64 cell_size = cs_prev_tensor->dim_size(1); variable
841 const int64 cell_size = cs_prev_tensor->dim_size(1); variable
851 << batch_size << ", cell_size=" << cell_size; variable
1066 const int64 cell_size = w_tensor->dim_size(1) \/ 4; variable
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| lstm.cc | 274 TfLiteIntArray* cell_size = TfLiteIntArrayCreate(2); local
275 cell_size->data[0] = n_batch;
276 cell_size->data[1] = n_cell;
278 context->ResizeTensor(context, cell_state, cell_size));
|
| unidirectional_sequence_lstm.cc | 276 TfLiteIntArray* cell_size = TfLiteIntArrayCreate(2); local
277 cell_size->data[0] = n_batch;
278 cell_size->data[1] = n_cell;
280 context->ResizeTensor(context, cell_state, cell_size));
|
| /bionic/libc/kernel/uapi/linux/ |
| pkt_sched.h | 437 __u32 cell_size; member in struct:tc_netem_rate
|
| /external/iproute2/include/uapi/linux/ |
| pkt_sched.h | 571 __u32 cell_size; member in struct:tc_netem_rate
|
| /external/kernel-headers/original/uapi/linux/ |
| pkt_sched.h | 575 __u32 cell_size; member in struct:tc_netem_rate
|
