| /external/federated-compute/fcp/tensorflow/ |
| D | tf_py_smoke_test.py | 26 def matmul(a, b): function
|
| /external/pytorch/torch/_numpy/ |
| D | _binary_ufuncs_impl.py | 54 def matmul(x, y): function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | tf_matmul_test.py | 23 def matmul(): function
|
| /external/executorch/backends/qualcomm/_passes/ |
| D | convert_bmm_to_matmul.py | 26 matmul = exir_ops.edge.aten.matmul.default variable in ConvertBmmToMatmul
|
| /external/pytorch/torch/ao/nn/quantized/modules/ |
| 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
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
|
| 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
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_fission.cc | 52 auto matmul = [&]() -> mlir::TF::MatMulOp { in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| 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
|
| D | c_api_remote_test.cc | 77 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute() local
|
| D | c_api_remote_test_util.cc | 137 TFE_Op* matmul = nullptr; in TestRemoteExecuteSilentCopies() local
|
| 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
|
| /external/pytorch/torch/utils/benchmark/examples/ |
| D | compare.py | 44 def matmul(self, *args, **kwargs): member in FauxTorch
|
| /external/pytorch/torch/ao/quantization/experimental/ |
| D | linear.py | 109 def matmul(self, decomposed_weight, activation): member in LinearAPoT
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | frozen_linear_transpose.cpp | 54 Node* matmul = nullptr; in replace_linear_with_matmul() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 79 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph() local
|
| /external/pytorch/torch/ |
| D | _linalg_utils.py | 32 def matmul(A: Optional[Tensor], B: Tensor) -> Tensor: function
|
| /external/pytorch/torch/linalg/ |
| D | __init__.py | 1553 matmul = _add_docstr(_linalg.linalg_matmul, r""" variable
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 228 def matmul(x, y, **kwargs): # pylint: disable=invalid-name function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | TestUtil.java | 74 public static <T> Output<T> matmul( in matmul() method in TestUtil
|
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | MatMul.cpp | 260 void matmul(ComputeGraph& graph, const std::vector<ValueRef>& args) { in matmul() function
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 266 foo::bar::MatMulComp matmul; in TEST() local
337 foo::bar::MatMulComp matmul( in TEST() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/detail/ |
| D | Matmul.cpp | 14 sycl::event matmul( in matmul() function
|