Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/ops/
Drnn_ops.cc43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anona90d3fa60102() local
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anona90d3fa60202() local
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anona90d3fa60302() local
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anona90d3fa60402() local
190 DimensionHandle cell_size; in __anona90d3fa60502() local
229 DimensionHandle cell_size; in __anona90d3fa60602() 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_t cell_size = h_prev_tensor->dim_size(1); in Compute() local
209 const int64_t cell_size = h_prev_tensor->dim_size(1); in Compute() local
Dlstm_ops.cc333 const int64_t cell_size = cs_prev_tensor->dim_size(1); in Compute() local
574 const int64_t cell_size = cs_prev_tensor->dim_size(1); in Compute() local
920 const int64_t cell_size = cs_prev_tensor->dim_size(1); in Compute() local
1165 const int64_t cell_size = w_tensor->dim_size(1) / 4; in Compute() local
/external/ComputeLibrary/tests/datasets/
DHOGDescriptorDataset.h87Size2D cell_size, Size2D block_size, Size2D detection_window_size, Size2D block_stride, in add_config()
/external/libnl/lib/route/
Dtc.c671 int rtnl_tc_calc_cell_log(int cell_size) in rtnl_tc_calc_cell_log()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream_executor_pimpl.cc376 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
Ddnn.h2039 int cell_size, int batch_size, in createRnnDescriptor()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1235 int hidden_size, int input_size, int cell_size, in CudnnRnnDescriptor()
1265 int cell_size, int batch_size, cudnnRNNInputMode_t input_mode, in Create()
1383 int cell_size() const { return cell_size_; } in cell_size() function in stream_executor::gpu::CudnnRnnDescriptor
1767 int cell_size = 0; member
2414 int num_layers, int hidden_size, int input_size, int cell_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/bpftool/include/uapi/linux/
Dpkt_sched.h619 __u32 cell_size; member
/external/kernel-headers/original/uapi/linux/
Dpkt_sched.h639 __u32 cell_size; member
/external/libbpf/include/uapi/linux/
Dpkt_sched.h619 __u32 cell_size; member
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2570 int num_layers, int hidden_size, int input_size, int cell_size, in createRnnDescriptor()
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h72403 u32 cell_size; member
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h105056 u32 cell_size; member
Dvmlinux.h105056 u32 cell_size; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h88151 u32 cell_size; member
Dvmlinux_510.h88151 u32 cell_size; member
/external/bcc/libbpf-tools/x86/
Dvmlinux_505.h104973 u32 cell_size; member
Dvmlinux.h104973 u32 cell_size; member