Searched refs:orig_node1 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2029 static void CopyAttrsFromPadAndConv2D(const Node* orig_node1,
2032 static void CopyAttrsFromPadAndFusedConv2D(const Node* orig_node1,
2667 void MklLayoutRewritePass::CopyAttrsFromPadAndConv2D(const Node* orig_node1, in CopyAttrsFromPadAndConv2D() argument
2680 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "T", &T)); in CopyAttrsFromPadAndConv2D()
2681 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "strides", &strides)); in CopyAttrsFromPadAndConv2D()
2682 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "dilations", &dilations)); in CopyAttrsFromPadAndConv2D()
2683 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "padding", &padding)); in CopyAttrsFromPadAndConv2D()
2684 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "data_format", &data_format)); in CopyAttrsFromPadAndConv2D()
2686 GetNodeAttr(orig_node1->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in CopyAttrsFromPadAndConv2D()
|