Searched refs:fused_conv2d (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 466 void CopyConv2DAttributes(const NodeDef* conv2d, NodeDef* fused_conv2d) { in CopyConv2DAttributes() argument
467 auto* attr = fused_conv2d->mutable_attr(); in CopyConv2DAttributes()
479 NodeDef* fused_conv2d, const absl::Span<const absl::string_view> fused_ops, in SetFusedConv2DAttributes() argument
481 auto* attr = fused_conv2d->mutable_attr(); in SetFusedConv2DAttributes()
498 NodeDef* fused_conv2d = optimized_graph->add_node(); in AddFusedConv2DNode() local
499 fused_conv2d->set_op(kFusedConv2D); in AddFusedConv2DNode()
500 fused_conv2d->set_name(matched.bias_add->name()); in AddFusedConv2DNode()
501 fused_conv2d->set_device(matched.conv2d->device()); in AddFusedConv2DNode()
502 fused_conv2d->add_input(matched.conv2d->input(0)); // 0: input in AddFusedConv2DNode()
503 fused_conv2d->add_input(matched.conv2d->input(1)); // 1: filter in AddFusedConv2DNode()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 267 csinfo_.fused_conv2d = "_FusedConv2D"; in MklLayoutRewritePass()
401 rinfo_.push_back({csinfo_.fused_conv2d, csinfo_.mkl_fused_conv2d, in MklLayoutRewritePass()
550 minfo_.push_back({csinfo_.pad, csinfo_.fused_conv2d, in MklLayoutRewritePass()
699 string fused_conv2d; member
964 e->dst()->type_string() == csinfo_.fused_conv2d) { in GetPadOrFusedConv2D()
971 DCHECK_EQ(m->type_string(), csinfo_.fused_conv2d); in GetPadOrFusedConv2D()
2174 const Node* fused_conv2d, const Node* pad, NodeBuilder* nb, in CopyAttrsFromPadAndFusedConv2D() argument
2187 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "T", &T)); in CopyAttrsFromPadAndFusedConv2D()
2188 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "num_args", &num_args)); in CopyAttrsFromPadAndFusedConv2D()
2189 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "strides", &strides)); in CopyAttrsFromPadAndFusedConv2D()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 728 NodeDef fused_conv2d; in RunFusedConv2DOp() local
738 .Finalize(&fused_conv2d)); in RunFusedConv2DOp()
740 RunAndFetch(root, fused_conv2d.name(), output, allow_gpu_device, in RunFusedConv2DOp()
741 &fused_conv2d); in RunFusedConv2DOp()
|