Home
last modified time | relevance | path

Searched defs:cell_size (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Drnn_ops.cc43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon264e79e10102() local
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon264e79e10202() local
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon264e79e10302() local
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon264e79e10402() local
190 DimensionHandle cell_size; in __anon264e79e10502() local
229 DimensionHandle cell_size; in __anon264e79e10602() local
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.h30 constexpr int gate_c_offset(GateLayout gate_layout, int cell_size) { in gate_c_offset()
34 constexpr int gate_f_offset(GateLayout gate_layout, int cell_size) { in gate_f_offset()
114 int cell_size() const { return cell_size_; } in cell_size() function
Dlstm_ops_gpu.cu.cc89 const int batch_size, const int cell_size) { in lstm_gates()
200 const int batch_size, const int cell_size, in concat_xh()
238 typename TTypes<T>::Matrix h, int batch_size, int cell_size, in LSTMBlockCellFpropWithCUDA()
310 const int batch_size, const int cell_size, const bool use_peephole) { in lstm_gates_bprop()
379 typename TTypes<T>::Vec wco_grad, const int batch_size, const int cell_size, in LSTMBlockCellBpropWithCUDA()
Dgru_ops.cc54 const int64 cell_size = h_prev_tensor->dim_size(1); in Compute() local
209 const int64 cell_size = h_prev_tensor->dim_size(1); in Compute() local
Dlstm_ops.cc333 const int64 cell_size = cs_prev_tensor->dim_size(1); in Compute() local
515 const int64 cell_size = cs_prev_tensor->dim_size(1); in Compute() local
861 const int64 cell_size = cs_prev_tensor->dim_size(1); in Compute() local
1095 const int64 cell_size = w_tensor->dim_size(1) / 4; in Compute() local
/external/libnl/lib/route/
Dtc.c678 int rtnl_tc_calc_cell_log(int cell_size) in rtnl_tc_calc_cell_log()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1060 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor()
1090 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create()
1205 int cell_size() const { return cell_size_; } in cell_size() function in stream_executor::gpu::CudnnRnnDescriptor
1592 int cell_size = 0; member
2071 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc366 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
Ddnn.h2111 int cell_size, int batch_size, in createRnnDescriptor()
/external/iproute2/include/uapi/linux/
Dpkt_sched.h571 __u32 cell_size; member
/external/libnl/include/linux-private/linux/
Dpkt_sched.h591 __u32 cell_size; member
/external/kernel-headers/original/uapi/linux/
Dpkt_sched.h639 __u32 cell_size; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2516 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()