Searched refs:kMklOpLabel (Results 1 – 21 of 21) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_relu_op.cc | 1071 .Label(mkl_op_registry::kMklOpLabel), \ 1076 .Label(mkl_op_registry::kMklOpLabel), \ 1085 .Label(mkl_op_registry::kMklOpLabel), \ 1090 .Label(mkl_op_registry::kMklOpLabel), \ 1098 .Label(mkl_op_registry::kMklOpLabel), \ 1103 .Label(mkl_op_registry::kMklOpLabel), \ 1111 .Label(mkl_op_registry::kMklOpLabel), \ 1116 .Label(mkl_op_registry::kMklOpLabel), \ 1124 .Label(mkl_op_registry::kMklOpLabel), \ 1129 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_reshape_op.cc | 256 .Label(mkl_op_registry::kMklOpLabel), \ 263 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_conv_ops.cc | 843 .Label(mkl_op_registry::kMklOpLabel), 848 .Label(mkl_op_registry::kMklOpLabel), 2091 .Label(mkl_op_registry::kMklOpLabel), \ 2097 .Label(mkl_op_registry::kMklOpLabel), \ 2103 .Label(mkl_op_registry::kMklOpLabel), \ 2109 .Label(mkl_op_registry::kMklOpLabel), \ 2116 .Label(mkl_op_registry::kMklOpLabel), \ 2123 .Label(mkl_op_registry::kMklOpLabel), \ 2132 .Label(mkl_op_registry::kMklOpLabel), \ 2145 .Label(mkl_op_registry::kMklOpLabel), \ [all …]
|
D | mkl_conv_grad_filter_ops.cc | 707 .Label(mkl_op_registry::kMklOpLabel), \ 713 .Label(mkl_op_registry::kMklOpLabel), \ 719 .Label(mkl_op_registry::kMklOpLabel), \ 724 .Label(mkl_op_registry::kMklOpLabel), \ 730 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_identity_op.cc | 59 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_avgpooling_op.cc | 707 .Label(mkl_op_registry::kMklOpLabel), 713 .Label(mkl_op_registry::kMklOpLabel), 721 .Label(mkl_op_registry::kMklOpLabel), 739 .Label(mkl_op_registry::kMklOpLabel),
|
D | mkl_cwise_ops_common.cc | 67 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_conv_grad_input_ops.cc | 564 .Label(mkl_op_registry::kMklOpLabel), \ 569 .Label(mkl_op_registry::kMklOpLabel), \ 574 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_tfconv_op.h | 140 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_maxpooling_op.cc | 869 .Label(mkl_op_registry::kMklOpLabel), 875 .Label(mkl_op_registry::kMklOpLabel), 883 .Label(mkl_op_registry::kMklOpLabel), 901 .Label(mkl_op_registry::kMklOpLabel),
|
D | mkl_softmax_op.cc | 189 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_input_conversion_op.cc | 302 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_concat_op.cc | 542 .Label(mkl_op_registry::kMklOpLabel), \ 549 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_conv_grad_bias_ops.cc | 262 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_aggregate_ops.cc | 258 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_lrn_op.cc | 678 .Label(mkl_op_registry::kMklOpLabel), \ 683 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_slice_op.cc | 484 .Label(mkl_op_registry::kMklOpLabel), \
|
D | mkl_fused_batch_norm_op.cc | 1119 .Label(mkl_op_registry::kMklOpLabel), \ 1128 .Label(mkl_op_registry::kMklOpLabel), \
|
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_graph_util.h | 76 static const char* kMklOpLabel = "MklOp"; variable
|
D | mkl_tfconversion_pass.cc | 192 conversion_node->AddAttr("_kernel", mkl_op_registry::kMklOpLabel); in InsertConversionNodeOnEdge() 277 conversion_node->AddAttr("_kernel", mkl_op_registry::kMklOpLabel); in InsertInputConversionNode()
|
D | mkl_layout_pass.cc | 3187 nb.Attr("_kernel", mkl_op_registry::kMklOpLabel); in RewriteNode()
|