Home
last modified time | relevance | path

Searched refs:fused_conv2d (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc965 void CopyConv2DAttributes(const NodeDef& conv2d, NodeDef* fused_conv2d, in IsCpuCompatibleDataType() argument
969 auto* attr = fused_conv2d->mutable_attr(); in IsCpuCompatibleDataType()
1172 NodeDef fused_conv2d; in IsCpuCompatibleDataType() local
1173 fused_conv2d.set_name(contraction.name()); in IsCpuCompatibleDataType()
1174 fused_conv2d.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1175 fused_conv2d.set_device(contraction.device()); in IsCpuCompatibleDataType()
1176 fused_conv2d.add_input(contraction.input(0)); // 0: input in IsCpuCompatibleDataType()
1177 fused_conv2d.add_input(contraction.input(1)); // 1: filter in IsCpuCompatibleDataType()
1178 fused_conv2d.add_input(bias_add.input(1)); // 2: bias in IsCpuCompatibleDataType()
1180 CopyConv2DAttributes(contraction, &fused_conv2d); in IsCpuCompatibleDataType()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc276 csinfo_.fused_conv2d = "_FusedConv2D"; in MklLayoutRewritePass()
488 rinfo_.push_back({csinfo_.fused_conv2d, in MklLayoutRewritePass()
715 minfo_.push_back({csinfo_.pad, csinfo_.fused_conv2d, in MklLayoutRewritePass()
935 string fused_conv2d; member
1233 e->dst()->type_string() == csinfo_.fused_conv2d) { in GetPadOrFusedConv2D()
1240 DCHECK_EQ(m->type_string(), csinfo_.fused_conv2d); in GetPadOrFusedConv2D()
2701 const Node* fused_conv2d, const Node* pad, NodeBuilder* nb, in CopyAttrsFromPadAndFusedConv2D() argument
2715 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "T", &T)); in CopyAttrsFromPadAndFusedConv2D()
2716 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "num_args", &num_args)); in CopyAttrsFromPadAndFusedConv2D()
2717 TF_CHECK_OK(GetNodeAttr(fused_conv2d->def(), "strides", &strides)); in CopyAttrsFromPadAndFusedConv2D()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc763 NodeDef fused_conv2d; in RunFusedConv2DOp() local
774 .Finalize(&fused_conv2d)); in RunFusedConv2DOp()
776 RunAndFetch(root, fused_conv2d.name(), output, allow_gpu_device, in RunFusedConv2DOp()
777 &fused_conv2d); in RunFusedConv2DOp()