HomeSort by relevance Sort by last modified time
    Searched refs:w_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/gan/python/features/python/
spectral_normalization_impl.py 50 def compute_spectral_norm(w_tensor, power_iteration_rounds=1, name=None):
54 w_tensor: The weight matrix whose spectral norm should be computed.
69 w = array_ops.reshape(w_tensor, (-1, w_tensor.get_shape()[-1]))
267 w_tensor = getter(name, *args, **kwargs)
269 if len(w_tensor.get_shape()) < 2:
274 w_tensor,
  /external/tensorflow/tensorflow/contrib/rnn/kernels/
lstm_ops.cc 253 const Tensor* w_tensor = nullptr; variable
254 OP_REQUIRES_OK(ctx, ctx->input("w", &w_tensor));
291 OP_REQUIRES(ctx, w_tensor->dim_size(0) == input_size + cell_size,
294 w_tensor->dim_size(0), " vs. ", input_size + cell_size));
295 OP_REQUIRES(ctx, w_tensor->dim_size(1) == cell_size * 4,
297 "w.dim_size(1) != cell_size * 4: ", w_tensor->dim_size(1),
360 h_prev_tensor->matrix<T>(), w_tensor->matrix<T>(), wci_tensor->vec<T>(),
434 const Tensor* w_tensor = nullptr; variable
435 OP_REQUIRES_OK(ctx, ctx->input("w", &w_tensor));
496 OP_REQUIRES(ctx, w_tensor->dim_size(0) == input_size + cell_size
874 const Tensor* w_tensor = nullptr; variable
1073 const Tensor* w_tensor = nullptr; variable
    [all...]

Completed in 260 milliseconds