Home
last modified time | relevance | path

Searched defs:all_weights (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp383 std::vector<Tensor> RNNImplBase<Derived>::all_weights() const { in all_weights() function in torch::nn::detail::RNNImplBase
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Drnn.py452 def all_weights(self) -> List[List[Parameter]]: member in RNNBase
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py4270 all_weights, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc4148 bool all_weights = true; in TEST_P() local