Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/ops/
Drnn_ops.cc43 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon4a6db9170102() local
74 DimensionHandle cell_size = c->Dim(h_prev, 1); in __anon4a6db9170202() local
111 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4a6db9170302() local
149 DimensionHandle cell_size = c->Dim(cs_prev, 1); in __anon4a6db9170402() local
190 DimensionHandle cell_size; in __anon4a6db9170502() local
229 DimensionHandle cell_size; in __anon4a6db9170602() 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/sdv/vsomeip/third_party/boost/numeric/odeint/examples/
Dmolecular_dynamics_cells.cpp119 double x_max , y_max , cell_size; member
209 double xmax = 100.0 , double ymax = 100.0 , double cell_size = 2.0 ) in md_system_bs()
318 double xmax = 100.0 , double ymax = 100.0 , double cell_size = 2.0 ) in make_md_system_bs()
/external/libnl/lib/route/
Dtc.c676 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/pytorch/aten/src/ATen/native/cudnn/
DRNN.cpp1458 auto cell_size = _cell_size(fn.rnn, fn.tensors); in _cudnn_rnn() local
1738 auto cell_size = _cell_size(fn.rnn, fn.tensors); in _cudnn_rnn_backward_input() local
/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/libbpf/include/uapi/linux/
Dpkt_sched.h510 __u32 cell_size; member
/external/bpftool/include/uapi/linux/
Dpkt_sched.h510 __u32 cell_size; member
/external/kernel-headers/original/uapi/linux/
Dpkt_sched.h531 __u32 cell_size; member
/external/libnl/include/linux-private/linux/
Dpkt_sched.h639 __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.h26014 u32 cell_size; member
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_600.h105773 u32 cell_size; member
Dvmlinux.h105773 u32 cell_size; member
/external/bcc/libbpf-tools/loongarch/
Dvmlinux_602.h98603 u32 cell_size; member
Dvmlinux.h98603 u32 cell_size; member
/external/bcc/libbpf-tools/x86/
Dvmlinux.h119018 u32 cell_size; member
Dvmlinux_518.h119018 u32 cell_size; member

12