- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for _FusedMatMul (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
if (!element_ty.isF32() && !element_ty.isBF16()) { (void)rewriter.notifyMatchFailure(matmul, [&](Diagnostic &diag) { diag << "supported data types for _FusedMatMul are float and bfloat16, " << " but got " << element_ty; }); return false; } // FusedMatMul kernel does not support grad_a/grad_b attrs if ((matmul->hasAttr("grad_a") &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/tf_to_corert_pipeline.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>; TF_DerivedOperandTypeListAttr TArgs = TF_DerivedOperandTypeListAttr<2>; } def TF__FusedMatMulOp : TF_Op<"_FusedMatMul", [Pure, TF_SameOperandsAndResultElementTypeResolveRef]> { let summary = [{ Performs a MatMul followed by a specified series of operations. }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)