Home
last modified time | relevance | path

Searched refs:p_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/v1_compat_tests/
Dgradient_checker_test.py103 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,
/external/tensorflow/tensorflow/python/ops/
Dgradient_checker_v2_test.py126 p_shape = (4, 2)
129 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
145 p_shape = (8, 2)
148 np.arange(p_size).astype(np.float), shape=p_shape, name="p")
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.h246 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size_}); in operator()
250 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape); in operator()
271 di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in operator()
272 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in operator()
Dlstm_ops_gpu.cu.cc396 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell_size}); in LSTMBlockCellBpropWithCUDA() local
399 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in LSTMBlockCellBpropWithCUDA()
400 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithCUDA()
Dlstm_ops.cc70 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()
91 auto f_peep = cs_prev * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
116 auto o_peep = cs * wco.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellFpropWithEigen()
153 Eigen::array<Eigen::DenseIndex, 2> p_shape({1, cell.cell_size()}); in LSTMBlockCellBpropWithEigen() local
157 dcs + do_ * wco.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithEigen()
179 cs_prev_grad + di * wci.reshape(p_shape).broadcast(p_broadcast_shape) + in LSTMBlockCellBpropWithEigen()
180 df * wcf.reshape(p_shape).broadcast(p_broadcast_shape); in LSTMBlockCellBpropWithEigen()
/external/tensorflow/tensorflow/core/kernels/
Dctc_decoder_ops.cc134 Tensor* p_shape = nullptr; in StoreAllDecodedSequences() local
143 s = decoded_shape->allocate(p, TensorShape({2}), &p_shape); in StoreAllDecodedSequences()
148 auto shape_t = p_shape->vec<int64>(); in StoreAllDecodedSequences()
/external/tensorflow/tensorflow/python/keras/applications/
Dnasnet.py556 p_shape = backend.int_shape(p)
562 elif p_shape[img_dim] != ip_shape[img_dim]:
601 elif p_shape[channel_dim] != filters:
/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops.cc197 ShapeHandle p_shape; in LuShapeFn() local
200 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Vector(n), &p_shape)); in LuShapeFn()
203 c->set_output(1, p_shape); in LuShapeFn()