Searched refs:fused_matmul (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_test.cc | 172 NodeDef fused_matmul; in RunFusedMatMulOp() local
182 .Finalize(&fused_matmul)); in RunFusedMatMulOp()
184 RunAndFetch(root, fused_matmul.name(), output, allow_gpu_device, in RunFusedMatMulOp()
185 &fused_matmul); in RunFusedMatMulOp()
208 Tensor fused_matmul; in VerifyBiasAddTensorsNear() local
211 run_fused(lhs, rhs, bias, &fused_matmul); in VerifyBiasAddTensorsNear()
213 ASSERT_EQ(matmul.dtype(), fused_matmul.dtype()); in VerifyBiasAddTensorsNear()
214 ASSERT_EQ(matmul.shape(), fused_matmul.shape()); in VerifyBiasAddTensorsNear()
216 test::ExpectClose(matmul, fused_matmul, /*atol=*/1e-5); in VerifyBiasAddTensorsNear()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 278 csinfo_.fused_matmul = "_FusedMatMul"; in MklLayoutRewritePass()
498 rinfo_.push_back({csinfo_.fused_matmul, in MklLayoutRewritePass()
937 string fused_matmul; member
3751 n->type_string() != csinfo_.fused_matmul && in CheckForNodeRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1033 void CopyMatMulAttributes(const NodeDef& matmul, NodeDef* fused_matmul, in IsCpuCompatibleDataType() argument
1037 auto* attr = fused_matmul->mutable_attr(); in IsCpuCompatibleDataType()
|