Searched refs:kMklOpLabel (Results 1 - 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
H A Dmkl_graph_util.h77 static const char* kMklOpLabel = "MklOp"; member in namespace:tensorflow::mkl_op_registry
H A Dmkl_tfconversion_pass.cc188 conversion_node->AddAttr("_kernel", mkl_op_registry::kMklOpLabel);
269 conversion_node->AddAttr("_kernel", mkl_op_registry::kMklOpLabel);
H A Dmkl_layout_pass.cc1836 new_node->AddAttr("_kernel", mkl_op_registry::kMklOpLabel);
1984 nb.Attr("_kernel", mkl_op_registry::kMklOpLabel);
4094 nb.Attr("_kernel", mkl_op_registry::kMklOpLabel);
/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_identity_op.cc90 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_cwise_ops_common.cc67 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_softmax_op.cc153 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_conv_grad_filter_ops.cc431 .Label(mkl_op_registry::kMklOpLabel), \
641 .Label(mkl_op_registry::kMklOpLabel), \
647 .Label(mkl_op_registry::kMklOpLabel), \
652 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_reshape_op.cc356 .Label(mkl_op_registry::kMklOpLabel), \
363 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_relu_op.cc843 .Label(mkl_op_registry::kMklOpLabel), \
848 .Label(mkl_op_registry::kMklOpLabel), \
859 .Label(mkl_op_registry::kMklOpLabel), \
864 .Label(mkl_op_registry::kMklOpLabel), \
872 .Label(mkl_op_registry::kMklOpLabel), \
877 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_conv_grad_bias_ops.cc258 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_tfconv_op.h199 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_conv_ops.cc759 .Label(mkl_op_registry::kMklOpLabel), \
764 .Label(mkl_op_registry::kMklOpLabel), \
769 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_conv_grad_input_ops.cc502 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_input_conversion_op.cc497 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_avgpooling_op.cc707 .Label(mkl_op_registry::kMklOpLabel),
713 .Label(mkl_op_registry::kMklOpLabel),
H A Dmkl_concat_op.cc818 .Label(mkl_op_registry::kMklOpLabel), \
825 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_maxpooling_op.cc801 .Label(mkl_op_registry::kMklOpLabel),
807 .Label(mkl_op_registry::kMklOpLabel),
H A Dmkl_aggregate_ops.cc501 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_fused_batch_norm_op.cc1358 .Label(mkl_op_registry::kMklOpLabel), \
1367 .Label(mkl_op_registry::kMklOpLabel), \
H A Dmkl_lrn_op.cc1352 .Label(mkl_op_registry::kMklOpLabel), \
1357 .Label(mkl_op_registry::kMklOpLabel), \

Completed in 904 milliseconds