| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_ops_test.cc | 150 const std::vector<string>& fused_ops, const FusedGraphRunner& run_default, in VerifyFusedTensorsClose()
177 const std::vector<string>& fused_ops, in VerifyFusedMatrixClose()
219 const std::vector<string>& fused_ops, Tensor* output, in RunConv2DUnfused()
285 const std::vector<string>& fused_ops, Tensor* output, in RunMklFusedConv2DOp()
324 const std::vector<string>& fused_ops, in VerifyFusedConv2D()
339 const Tensor& bias_data, const std::vector<string>& fused_ops, in VerifyFusedConv2D()
340 Tensor* out, const int padding) { in VerifyFusedConv2D()
537 const std::vector<string>& fused_ops, in RunDepthwiseConv2DUnfused()
579 const std::vector<string>& fused_ops, in RunMklFusedDepthwiseConv2DOp()
615 const std::vector<string>& fused_ops, in VerifyFusedDepthwiseConv2D()
[all …]
|
| H A D | mkl_conv_ops.cc | 1369 std::vector<string> fused_ops; in MklFusedConvOp() local
1564 std::vector<string> fused_ops; in MklFusedDepthwiseConvOp() local
2129 std::vector<string> fused_ops; in MklFusedConv3DOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | remapper_test.cc | 702 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
847 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
972 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
1065 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
1131 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
1197 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
1259 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
1471 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
1567 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
1673 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
[all …]
|
| H A D | mkl_remapper_test.cc | 148 const auto fused_ops = node.attr().at("fused_ops").list().s(); in FuseConv2DWithBiasAndAddNOrAdd() local
517 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
609 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RelpaceAddWithBiasAddDepthwiseConv2D() local
623 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RelpaceAddWithBiasAddDepthwiseConv2D() local
713 const auto fused_ops = node.attr().at("fused_ops").list().s(); in RunTest() local
813 const auto fused_ops = node.attr().at("fused_ops").list().s(); in VerifyFused() local
|
| H A D | remapper.cc | 1291 auto fused_ops = matmul_node->attr().at("fused_ops").list().s(); in FindMatMulBiasAddAndGelu() local
2199 const absl::Span<const absl::string_view> fused_ops, in SetFusedOpAttributes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_eigen_output_kernels.cc | 31 std::vector<string> fused_ops; in InitializeFusedComputation() local
|
| H A D | fused_eigen_output_kernels.h | 59 std::vector<string> fused_ops; member
|
| H A D | conv_ops_benchmark_test.cc | 231 const std::vector<string>& fused_ops = {}, in FusedConv2DWithBias()
274 int out_depth, const std::vector<string>& fused_ops = {}, in FusedConv2DWithBatchNorm()
|
| H A D | matmul_op_test.cc | 152 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp()
|
| H A D | conv_ops_test.cc | 684 const std::vector<string>& fused_ops, in RunFusedConv2DOp()
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 674 fused_ops = [ variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | fused_kernel_matcher.cc | 154 SmallVector<Attribute, 2> fused_ops{StringAttr::get( in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | mkl_layout_pass.cc | 1740 std::vector<string> fused_ops; in FusedConv2DRewrite() local
1776 std::vector<string> fused_ops; in FusedDepthwiseConv2DRewrite() local
2687 std::vector<string> fused_ops; in CopyAttrsFromPadAndFusedConv2D() local
|