Searched refs:batch_cell_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 936 TensorShape batch_cell_shape({timelen, batch_size, cell_size}); in Compute() local
938 OP_REQUIRES_OK(ctx, ctx->allocate_output("i", batch_cell_shape, &i_out)); in Compute()
941 OP_REQUIRES_OK(ctx, ctx->allocate_output("cs", batch_cell_shape, &cs_out)); in Compute()
944 OP_REQUIRES_OK(ctx, ctx->allocate_output("f", batch_cell_shape, &f_out)); in Compute()
947 OP_REQUIRES_OK(ctx, ctx->allocate_output("o", batch_cell_shape, &o_out)); in Compute()
950 OP_REQUIRES_OK(ctx, ctx->allocate_output("ci", batch_cell_shape, &ci_out)); in Compute()
953 OP_REQUIRES_OK(ctx, ctx->allocate_output("co", batch_cell_shape, &co_out)); in Compute()
956 OP_REQUIRES_OK(ctx, ctx->allocate_output("h", batch_cell_shape, &h_out)); in Compute()
1179 TensorShape batch_cell_shape({batch_size, cell_size}); in Compute() local
1193 batch_cell_shape, &do_tensor)); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rnn_ops.cc | 76 ShapeHandle batch_cell_shape = c->Matrix(batch_size, cell_size); in __anon652ace880202() local
79 c->set_output(1, batch_cell_shape); in __anon652ace880202()
80 c->set_output(2, batch_cell_shape); in __anon652ace880202()
|