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

  /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
attention_wrapper.py 54 "safe_cumprod",
582 def safe_cumprod(x, *args, **kwargs): function
667 # safe_cumprod computes cumprod in logspace with numeric checks
668 cumprod_1mp_choose_i = safe_cumprod(1 - p_choose_i, axis=1, exclusive=True)
    [all...]
  /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
attention_wrapper_test.py 424 # Compute cumprod with safe_cumprod
425 safe_cumprod_output = wrapper.safe_cumprod(
    [all...]

Completed in 77 milliseconds