| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | c_api_test.cc | 67 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp() local
88 TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); in BM_Execute() local
428 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy() local
484 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices() local
756 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU() local
793 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error() local
865 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Type_Error() local
929 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithTracing() local
1293 TFE_Op* matmul = TFE_NewOp(ctx, "MatMulFunction", status); in BM_ExecuteFunction() local
|
| H A D | custom_device_test.cc | 49 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST() local
298 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST() local
|
| H A D | c_api_remote_test.cc | 77 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute() local
|
| H A D | c_api_remote_test_util.cc | 137 TFE_Op* matmul = nullptr; in TestRemoteExecuteSilentCopies() local
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | functional_modules.py | 96 def matmul(self, x: Tensor, y: Tensor) -> Tensor: member in FloatFunctional
160 def matmul(self, x: Tensor, y: Tensor) -> Tensor: member in FXFloatFunctional
284 def matmul(self, x: Tensor, y: Tensor) -> Tensor: member in QFunctional
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_mkldnn_contraction_kernel_test.cc | 140 Scalar matmul = matmul_result(i % m, i / m); in TEST() local
211 Scalar matmul = matmul_result(i, j); in TEST() local
|
| H A D | matmul_op_test.cc | 106 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBias() local
125 ops::MatMul matmul = ops::MatMul( in RunMatMulWithBiasAndActivation() local
207 Tensor matmul; in VerifyBiasAddTensorsNear() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | fused_kernel_matcher.cc | 290 bool AreFuseCompatible(MatMulOp matmul, BiasAddOp bias_add, in AreFuseCompatible()
303 bool IsDeviceCompatible(MatMulOp matmul, BiasAddOp bias_add, in IsDeviceCompatible()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | quantize_training_test.cc | 296 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
333 Node* matmul = test::graph::Matmul(graph, relu, identity, false, false); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/aot/tests/ |
| H A D | tfcompile_test.cc | 266 foo::bar::MatMulComp matmul; in TEST() local
337 foo::bar::MatMulComp matmul( in TEST() local
|
| /aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
| H A D | tf_py_smoke_test.py | 26 def matmul(a, b): function
|
| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _binary_ufuncs_impl.py | 54 def matmul(x, y): function
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | convert_bmm_to_matmul.py | 26 matmul = exir_ops.edge.aten.matmul.default variable in ConvertBmmToMatmul
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| H A D | tf_matmul_test.py | 23 def matmul(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| H A D | tf_cpurt_passes.cc | 355 mlir::linalg::MatmulOp matmul, mlir::PatternRewriter& rewriter) const { in matchAndRewrite()
520 auto matmul = [&]() -> mlir::TF::MatMulOp { in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_fission.cc | 52 auto matmul = [&]() -> mlir::TF::MatMulOp { in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/torch/utils/benchmark/examples/ |
| H A D | compare.py | 44 def matmul(self, *args, **kwargs): member in FauxTorch
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/experimental/ |
| H A D | linear.py | 109 def matmul(self, decomposed_weight, activation): member in LinearAPoT
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | frozen_linear_transpose.cpp | 54 Node* matmul = nullptr; in replace_linear_with_matmul() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | analytical_cost_estimator_test.cc | 79 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _linalg_utils.py | 32 def matmul(A: Optional[Tensor], B: Tensor) -> Tensor: function
|
| /aosp_15_r20/external/pytorch/torch/linalg/ |
| H A D | __init__.py | 1553 matmul = _add_docstr(_linalg.linalg_matmul, r""" variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | mkl_remapper_test.cc | 466 auto matmul = ops::MatMul(s.WithOpName("matmul"), input, filter); in TEST_F() local
667 auto matmul = ops::MatMul(s.WithOpName("matmul"), lhs, rhs); in RunTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| H A D | sparse_csr_matrix_grad.py | 228 def matmul(x, y, **kwargs): # pylint: disable=invalid-name function
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | MatMul.cpp | 260 void matmul(ComputeGraph& graph, const std::vector<ValueRef>& args) { in matmul() function
|