lstm_ops.cc | 262 const Tensor* b_tensor = nullptr; variable 263 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); 297 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, 299 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), 358 wcf_tensor->vec<T>(), wco_tensor->vec<T>(), b_tensor->vec<T>(), 441 const Tensor* b_tensor = nullptr; variable 442 OP_REQUIRES_OK(ctx, ctx->input("b", &b_tensor)); 500 OP_REQUIRES(ctx, b_tensor->dim_size(0) == cell_size * 4, 502 "b.dim_size(0) != cell_size * 4: ", b_tensor->dim_size(0), 641 wco_tensor->vec<T>(), b_tensor->vec<T>(), i_tensor->matrix<T>() 907 const Tensor* b_tensor = nullptr; variable 1081 const Tensor* b_tensor = nullptr; variable [all...] |