| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Blas.cpp | 175 const Tensor& batch2, in baddbmm_out()
236 const Tensor& batch2, in baddbmm_()
247 const Tensor& batch2, in baddbmm()
259 const Tensor& batch2, in addbmm_out()
301 const Tensor& batch2, in addbmm_()
311 const Tensor& batch2, in addbmm()
319 Tensor& bmm_out(const Tensor& self, const Tensor& batch2, Tensor& result) { in bmm_out()
339 Tensor bmm(const Tensor& self, const Tensor& batch2) { in bmm()
|
| /aosp_15_r20/external/sdk-platform-java/gax-java/gax/src/test/java/com/google/api/gax/rpc/ |
| H A D | BatchTest.java | 68 Batch<LabeledIntList, List<Integer>> batch2 = createBatch(request2); in testMerge() local
82 Batch<LabeledIntList, List<Integer>> batch2 = createBatch(request2); in testMergeStartEmpty() local
116 Batch<LabeledIntList, List<Integer>> batch2 = createBatch(request2); in testBatchMergerImpl() local
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | bmm_test.py | 28 def forward(self, batch1, batch2): argument
|
| H A D | add_test.py | 134 def forward(self, input_one, batch1, batch2): argument
|
| /aosp_15_r20/packages/modules/Bluetooth/android/app/tests/unit/src/com/android/bluetooth/opp/ |
| D | BluetoothOppServiceTest.java | 171 BluetoothOppBatch batch2 = new BluetoothOppBatch(mService, shareInfo2); in deleteShare_deleteShareAndCorrespondingBatch() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebra.cpp | 284 void common_checks_baddbmm_bmm(Meta& meta, const Tensor& batch1, const Tensor& batch2, const Scalar… in common_checks_baddbmm_bmm()
1549 …Tensor &result, const Tensor &self, const Tensor &batch1, const Tensor &batch2, const Scalar& beta… in addbmm_impl_()
1589 Tensor& addbmm_out(const Tensor& self, const Tensor& batch1, const Tensor& batch2, const Scalar& be… in addbmm_out()
1600 Tensor &addbmm_(Tensor& self, const Tensor& batch1, const Tensor& batch2, const Scalar& beta, const… in addbmm_()
1604 Tensor addbmm(const Tensor& self, const Tensor& batch1, const Tensor& batch2, const Scalar& beta, c… in addbmm()
1735 …_or_baddbmm_(const Tensor& self_or_result_, const Tensor& batch1, const Tensor& batch2, const Scal… in bmm_out_or_baddbmm_()
|
| /aosp_15_r20/external/perfetto/ui/src/trace_processor/ |
| H A D | query_result_unittest.ts | 251 const batch2 = protos.QueryResult.create({ constant
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 126 …const Tensor& input, const Tensor& batch1, const Tensor& batch2, const Scalar& beta, const Scalar&… in addbmm_decomp()
138 …const Tensor& input, const Tensor& batch1, const Tensor& batch2, const Scalar& beta, const Scalar&… in baddbmm_decomp()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Blas.cpp | 511 …l(const Tensor& result, const Tensor& self, const Tensor& batch1, const Tensor& batch2, const Scal… in baddbmm_out_cuda_impl()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2020 def meta_baddbmm(self, batch1, batch2, *, beta=1, alpha=1): argument
3129 def meta_addbmm(self, batch1, batch2, *, beta=1, alpha=1): argument
3686 def common_meta_baddbmm_bmm(batch1, batch2, is_bmm, self_baddbmm=None): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 401 const auto& batch2 = p_node->Input(2).toTensor(); in __anon5d9c3eb92702() local
2439 const auto& batch2 = p_node->Input(2).toTensor(); in __anon5d9c3eb9eb02() local
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 4954 def baddbmm(self, batch1, batch2, beta=1, alpha=1): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 5626 def baddbmm(g: jit_utils.GraphContext, self, batch1, batch2, beta, alpha): argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 7820 def forward(self, input, batch1, batch2): argument
7833 def forward(self, input, batch1, batch2, alpha, beta): argument
|