| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Blas.cpp | 174 const Tensor& batch1, in baddbmm_out()
235 const Tensor& batch1, in baddbmm_()
246 const Tensor& batch1, in baddbmm()
258 const Tensor& batch1, in addbmm_out()
300 const Tensor& batch1, in addbmm_()
310 const Tensor& batch1, in addbmm()
|
| /aosp_15_r20/external/sdk-platform-java/gax-java/gax/src/test/java/com/google/api/gax/rpc/ |
| H A D | BatchTest.java | 65 Batch<LabeledIntList, List<Integer>> batch1 = createBatch(request1); in testMerge() local
78 Batch<LabeledIntList, List<Integer>> batch1 = createBatch(request1, null); in testMergeStartEmpty() local
113 Batch<LabeledIntList, List<Integer>> batch1 = createBatch(request1); 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 | 170 BluetoothOppBatch batch1 = new BluetoothOppBatch(mService, shareInfo); 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 static inline void bmm_out_or_baddbmm_(const Tensor& self_or_result_, const Tensor& batch1, const T… in bmm_out_or_baddbmm_()
|
| /aosp_15_r20/external/perfetto/ui/src/trace_processor/ |
| H A D | query_result_unittest.ts | 241 const batch1 = 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 const Tensor& baddbmm_out_cuda_impl(const Tensor& result, const Tensor& self, const Tensor& batch1,… 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 | 400 const auto& batch1 = p_node->Input(1).toTensor(); in __anon5d9c3eb92702() local
2438 const auto& batch1 = p_node->Input(1).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
|