| lstm_ops.cc | 321 Tensor* o_tensor = nullptr; variable
324 TensorShape({batch_size, cell_size}), &o_tensor));
363 f_tensor->matrix<T>(), o_tensor->matrix<T>(), ci_tensor->matrix<T>(),
458 const Tensor* o_tensor = nullptr; variable
459 OP_REQUIRES_OK(ctx, ctx->input("o", &o_tensor));
537 OP_REQUIRES(ctx, o_tensor->dim_size(0) == batch_size,
539 "o.dim_size(0) != batch_size: ", o_tensor->dim_size(0),
541 OP_REQUIRES(ctx, o_tensor->dim_size(1) == cell_size,
543 "o.dim_size(1) != cell_size: ", o_tensor->dim_size(1),
647 cs_tensor->matrix<T>(), f_tensor->matrix<T>(), o_tensor->matrix<T>()
972 Tensor o_tensor = slicer.OutputSlice(o_out, t, "o_out"); variable
1229 const Tensor& o_tensor = slicer.InputSlice(*o_out, t, "o_out"); variable
[all...] |
