Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:h_prev_tensor (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/contrib/rnn/kernels/
gru_ops.cc	37 const Tensor* h_prev_tensor = nullptr; variable 38 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); 54 const int64 cell_size = h_prev_tensor->dim_size(1); 59 OP_REQUIRES(ctx, h_prev_tensor->dim_size(0) == batch_size, 61 h_prev_tensor->dim_size(0), " vs. ", 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), 152 ctx, device, x_tensor->matrix<T>(), h_prev_tensor->matrix<T>(), 180 const Tensor* h_prev_tensor = nullptr; variable 181 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor));     [all...]
lstm_ops.cc	250 const Tensor* h_prev_tensor = nullptr; variable 251 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); 282 OP_REQUIRES(ctx, h_prev_tensor->dim_size(0) == batch_size, 284 h_prev_tensor->dim_size(0), " vs. ", 286 OP_REQUIRES(ctx, h_prev_tensor->dim_size(1) == cell_size, 288 "h_prev.dims(1) != cell_size: ", h_prev_tensor->dim_size(1), 360 h_prev_tensor->matrix<T>(), w_tensor->matrix<T>(), wci_tensor->vec<T>(), 431 const Tensor* h_prev_tensor = nullptr; variable 432 OP_REQUIRES_OK(ctx, ctx->input("h_prev", &h_prev_tensor)); 487 OP_REQUIRES(ctx, h_prev_tensor->dim_size(0) == batch_size 861 const Tensor* h_prev_tensor = nullptr; variable 1070 const Tensor* h_prev_tensor = nullptr; variable     [all...]

Completed in 294 milliseconds