Home
last modified time | relevance | path

Searched defs:fused_ops (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc152 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
216 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
294 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
372 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
441 const auto fused_ops = node.attr().at("fused_ops").list().s(); in TEST_F() local
Dremapper.cc479 NodeDef* fused_conv2d, const absl::Span<const absl::string_view> fused_ops, in SetFusedConv2DAttributes()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_ops_test.cc173 const std::vector<string>& fused_ops, Tensor* output, in RunMklFusedConv2DOp()
597 const std::vector<string>& fused_ops, in RunMklFusedConv2DWithPadOp()
Dconv_ops_test.cc709 const std::vector<string>& fused_ops, Tensor* output, in RunFusedConv2DOp()
1233 const std::vector<string>& fused_ops = {}) { in FusedConv2DWithBias()
1265 int out_depth, const std::vector<string>& fused_ops = {}) { in FusedConv2DWithBatchNorm()
Dmkl_conv_ops.cc1465 std::vector<string> fused_ops; in MklFusedConvOp() local
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1321 std::vector<string> fused_ops; in FusedConv2DRewrite() local
2183 std::vector<string> fused_ops; in CopyAttrsFromPadAndFusedConv2D() local
2591 std::vector<string> fused_ops; in CopyAttrsFusedConv2D() local