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

  /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
loss.py 129 reduce_axis = [0] if average_across_batch else [1]
130 crossent = math_ops.reduce_sum(crossent, axis=reduce_axis)
131 total_size = math_ops.reduce_sum(weights, axis=reduce_axis)
134 reduce_axis = [0] if sum_over_batch else [1]
135 crossent = math_ops.reduce_sum(crossent, axis=reduce_axis)
137 math_ops.count_nonzero(weights, axis=reduce_axis),
  /external/tensorflow/tensorflow/python/ops/
nn_grad.py 944 reduce_axis = [0, 1, 2]
947 reduce_axis = [0, 2, 3]
950 mean_grad_y = math_ops.reduce_mean(grad_y, reduce_axis, keepdims=keepdims)
951 mean_x = math_ops.reduce_mean(x, reduce_axis, keepdims=keepdims)
954 reduce_axis,
959 grad_y * x_offset, axis=reduce_axis, keepdims=keepdims)
963 grad_y * x_offset, axis=reduce_axis, keepdims=keepdims)
966 grad_offset = math_ops.reduce_sum(grad_y, axis=reduce_axis)
970 reduce_axis = [0, 1, 2]
972 reduce_axis = [0, 2, 3
    [all...]
  /external/tensorflow/tensorflow/python/ops/ragged/
ragged_tensor.py     [all...]

Completed in 1229 milliseconds