Searched refs:orig_node1 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 1507 static void CopyAttrsFromPadAndConv2D(const Node* orig_node1,
1510 static void CopyAttrsFromPadAndFusedConv2D(const Node* orig_node1,
2140 void MklLayoutRewritePass::CopyAttrsFromPadAndConv2D(const Node* orig_node1, in CopyAttrsFromPadAndConv2D() argument
2153 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "T", &T)); in CopyAttrsFromPadAndConv2D()
2154 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "strides", &strides)); in CopyAttrsFromPadAndConv2D()
2155 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "dilations", &dilations)); in CopyAttrsFromPadAndConv2D()
2156 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "padding", &padding)); in CopyAttrsFromPadAndConv2D()
2157 TF_CHECK_OK(GetNodeAttr(orig_node1->def(), "data_format", &data_format)); in CopyAttrsFromPadAndConv2D()
2159 GetNodeAttr(orig_node1->def(), "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in CopyAttrsFromPadAndConv2D()
|