| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | tf_matmul_test.py | 23 def matmul(): function
|
| /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()
|
| D | unroll_batch_matmul.cc | 258 auto matmul = rewriter.create<TF::MatMulOp>(loc, matmul_type, in matchAndRewrite() local
|
| /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/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_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
|
| D | c_api_remote_test_util.cc | 137 TFE_Op* matmul = nullptr; in TestRemoteExecuteSilentCopies() local
|
| /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/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 79 auto matmul = ops::MatMul(s.WithOpName("matmul"), flat, w2); in CreateMiniGraph() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_grad.py | 228 def matmul(x, y, **kwargs): # pylint: disable=invalid-name function
|
| D | sparse_csr_matrix_ops.py | 141 def matmul(a, 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/python/cpython3/Lib/test/ |
| D | test_type_comments.py | 35 matmul = """\ variable
|
| /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/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_rewriter.cc | 306 HloInstruction *matmul) { in FuseVectorBiasAdd()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_algebra.py | 118 def matmul(lin_op_a, lin_op_b, name=None): function
|
| D | linear_operator_block_lower_triangular.py | 399 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperatorBlockLowerTriangular
|
| D | linear_operator_block_diag.py | 296 def matmul(self, x, adjoint=False, adjoint_arg=False, name="matmul"): member in LinearOperatorBlockDiag
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test_util.py | 179 def matmul(self, x): member in ModelTest._getMatMulModelWithSmallWeights.MatMulModelWithSmallWeights
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/ |
| D | quantize_model_test_base.py | 371 def matmul(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]: member in QuantizedModelTest._create_matmul_model.MatmulModel
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| 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
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| 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
|