Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp430 const Tensor& batch_sizes, in forward_helper()
518 auto batch_sizes = torch::Tensor(); in forward() local
535 const auto& batch_sizes = packed_input.batch_sizes(); in forward_with_packed_input() local
608 const Tensor& batch_sizes, in forward_helper()
669 auto batch_sizes = torch::Tensor(); in forward() local
686 const auto& batch_sizes = packed_input.batch_sizes(); in forward_with_packed_input() local
717 const Tensor& batch_sizes, in forward_helper()
766 auto batch_sizes = torch::Tensor(); in forward() local
782 const auto& batch_sizes = packed_input.batch_sizes(); in forward_with_packed_input() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp36 IntArrayRef batch_sizes, in mkldnn_rnn_layer()
68 at::IntArrayRef batch_sizes, in mkldnn_rnn_layer_backward()
95 at::IntArrayRef batch_sizes; member
224 IntArrayRef batch_sizes, in mkldnn_rnn_layer()
315 at::IntArrayRef batch_sizes, in mkldnn_rnn_layer_backward()
459 bool train, bool bidirectional, IntArrayRef batch_sizes) { in mkldnn_rnn()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPackedSequence.cpp64 int64_t * batch_sizes = batch_sizes_t.mutable_data_ptr<int64_t>(); in _pack_padded_sequence() local
129 int64_t * batch_sizes = batch_sizes_t.data_ptr<int64_t>(); in _pack_padded_sequence_backward_symint() local
146 int64_t * batch_sizes = batch_sizes_t.data_ptr<int64_t>(); in _pad_packed_sequence() local
H A DRNN.cpp133 Tensor batch_sizes; member
936 int64_t* batch_sizes = input.batch_sizes.data_ptr<int64_t>(); in operator ()() local
995 int64_t* batch_sizes = input.batch_sizes.data_ptr<int64_t>(); in operator ()() local
1400 const Tensor& batch_sizes, in quantized_gru_data_legacy()
1492 const Tensor& data, const Tensor& batch_sizes, TensorList hx, in lstm()
1749 const Tensor& batch_sizes, in quantized_lstm_data()
1780 const Tensor& batch_sizes, in quantized_lstm_data_legacy()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp43 …double dropout, bool train, bool bidirectional, IntArrayRef batch_sizes, const std::optional<Tenso… in miopen_rnn_backward()
134 std::vector<TensorDescriptor> rnn_descriptor_sequence(const Tensor& tensor, IntArrayRef batch_sizes… in rnn_descriptor_sequence()
159 IntArrayRef batch_sizes; member
763 …double dropout, bool train, bool bidirectional, IntArrayRef batch_sizes, const std::optional<Tenso… in miopen_rnn_backward()
830 …IntArrayRef batch_sizes { _batch_sizes.data_ptr<int64_t>(), static_cast<size_t>(_batch_sizes.size(… in _miopen_impl() local
898 const Tensor& data, const Tensor& batch_sizes, TensorList hx, in lstm_packed_miopen()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Drebatch_dataset_op.cc302 std::vector<int64_t> batch_sizes; in MakeDataset() local
320 std::vector<int64_t>&& batch_sizes, bool drop_remainder, in Dataset()
371 Node* batch_sizes = nullptr; in AsGraphDefInternal() local
H A Dauto_shard_dataset_op_test.cc338 Node* batch_sizes; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DRNN.cpp89 IntArrayRef batch_sizes, in _cudnn_rnn_backward()
288 IntArrayRef batch_sizes) { in rnn_descriptor_sequence()
315 const IntArrayRef batch_sizes, in rnn_descriptor_sequence()
424 IntArrayRef batch_sizes; member
2107 IntArrayRef batch_sizes, in _cudnn_rnn_backward()
2533 IntArrayRef batch_sizes{ in _cudnn_impl() local
2724 const Tensor& batch_sizes, in lstm_packed_cudnn()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp169 auto batch_sizes = ArrayRef<int64_t>( in batchedTensorInplaceForLoopFallback() local
303 auto batch_sizes = ArrayRef<int64_t>(some_sizes.begin(), some_sizes.begin() + num_batch_dims); in batchedTensorForLoopFallback() local
H A DLegacyVmapTransforms.cpp214 VmapDimVector batch_sizes(num_batch_dims, 1); in logicalToPhysical() local
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp174 auto batch_sizes = ArrayRef<int64_t>( in batchedTensorInplaceForLoopFallback() local
319 auto batch_sizes = ArrayRef<int64_t>(some_sizes.begin(), some_sizes.begin() + num_batch_dims); in batchedTensorForLoopFallback() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/
H A Drnn.h89 const Tensor& batch_sizes() const { in batch_sizes() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/
H A DAutocastRNN.cpp36 IntArrayRef batch_sizes, in _cudnn_rnn_cast_reflatten()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
H A Dhvp_test.py112 def _hvp_benchmark(self, hvp_fn, label, batch_sizes, argument
/aosp_15_r20/external/pytorch/benchmarks/transformer/
H A Dsdp.py277 batch_sizes, num_heads, max_seq_lens, embed_dims, dtypes, pad_percentages argument
/aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/
H A Ddistribute.py121 def __init__(self, input_dataset, batch_sizes, drop_remainder=False): argument
/aosp_15_r20/external/pytorch/torch/_decomp/
H A Ddecompositions.py2999 inp, hidden, params, has_biases, hidden_fn, batch_sizes, reverse=False argument
3239 batch_sizes, argument
3274 batch_sizes, argument
3345 def one_layer_lstm_data(inp, hidden, params, has_biases, batch_sizes, reverse=False): argument
3497 batch_sizes, argument
3548 batch_sizes, argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dpeephole.cpp308 Value* batch_sizes = n->outputs().at(1); in pushPackingPastRnn() local
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py4277 batch_sizes=None, argument
4527 batch_sizes, argument
4634 batch_sizes, argument
4724 batch_sizes, argument
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dnn_utils.cpp662 torch::Tensor batch_sizes = torch::empty({max_length}, torch::kInt64); in TEST_F() local
H A Ddataloader.cpp1641 const size_t batch_sizes[] = {5, 7}; in TEST() local
1798 const size_t batch_sizes[] = {17, 30}; in TEST() local
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py5932 batch_sizes, argument
5982 batch_sizes, argument
6179 batch_sizes, argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_torch.py3399 def make_arg(batch_sizes, n, dim, contig): argument
3561 def make_arg(batch_sizes, n, dim, contig): argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_modules.py2706 def make_packed_sequence(inp, batch_sizes): argument
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp7045 at::IntArrayRef batch_sizes, in mkldnn_rnn_layer_differentiable_backward()