Searched refs:p_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_test.py | 103 p_shape = (4, 2)
108 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
112 error = gradient_checker.compute_gradient_error(params, p_shape, y,
121 p_shape = (8, 2)
128 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
134 error = gradient_checker.compute_gradient_error(params, p_shape, y2,
|
| D | gradient_checker_v2_test.py | 76 p_shape = (4, 2)
79 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
93 p_shape = (8, 2)
96 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.h | 233 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size_}); in operator()
237 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape); in operator()
258 di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in operator()
259 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in operator()
|
| D | lstm_ops_gpu.cu.cc | 389 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size}); in LSTMBlockCellBpropWithCUDA() local
392 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in LSTMBlockCellBpropWithCUDA()
393 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithCUDA()
|
| D | lstm_ops.cc | 70 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellFpropWithEigen() local
75 auto i_peep = cs_prev * wci.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
89 auto f_peep = cs_prev * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
112 auto o_peep = cs * wco.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
149 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellBpropWithEigen() local
153 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithEigen()
173 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in LSTMBlockCellBpropWithEigen()
174 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithEigen()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 132 Tensor* p_shape = nullptr; in StoreAllDecodedSequences() local
141 s = decoded_shape->allocate(p, TensorShape({2}), &p_shape); in StoreAllDecodedSequences()
146 auto shape_t = p_shape->vec<int64>(); in StoreAllDecodedSequences()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 126 ShapeHandle p_shape; in LuShapeFn() local
129 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Vector(n), &p_shape)); in LuShapeFn()
132 c->set_output(1, p_shape); in LuShapeFn()
|