Searched refs:softmax_out (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | softmax_op_gpu.cu.cc | 143 Tensor* softmax_out = nullptr; in Compute() local
145 {0}, 0, logits_in_.shape(), &softmax_out)); in Compute()
153 softmax_out->shape(), &max_logits)); in Compute()
158 softmax_out->shape(), &sum_probs)); in Compute()
188 const_cast<T*>(softmax_out->flat<T>().data()), rows, cols, log_)); in Compute()
|
| D | softmax_op.cc | 67 Tensor* softmax_out = nullptr; in Compute() local
69 {0}, 0, logits_in.shape(), &softmax_out)); in Compute()
73 softmax_out->flat_inner_dims<T>(), log_); in Compute()
|