| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.cc | 33 using mkldnn::prop_kind;
44 context_.prop_kind = fwdParams.prop_kind; in Setup()
64 fwdParams.prop_kind, fwdParams.alg_kind, *context_.src_md, in Setup()
83 fwdParams.prop_kind == prop_kind::forward_training) { in Setup()
111 context_.prop_kind == in Execute()
112 prop_kind::forward_training) { // max pooling must have ws in Execute()
122 context_.prop_kind == in Execute()
123 prop_kind::forward_training) { // max pooling must have ws in Execute()
158 bwdParams.prop_kind, bwdParams.alg_kind, *context_.diff_src_md, in Setup()
|
| D | mkl_pooling_ops_common.h | 43 using mkldnn::prop_kind;
53 mkldnn::prop_kind prop_kind; member
58 mkldnn::algorithm alg_kind, mkldnn::prop_kind prop_kind) in MklPoolingParams()
66 prop_kind(prop_kind) {} in MklPoolingParams()
103 mkldnn::prop_kind prop_kind; member
195 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.prop_kind)); in CreateKey()
|
| D | mkl_softmax_op.cc | 31 using mkldnn::prop_kind;
130 auto softmax_fwd_desc = softmax_forward::desc(prop_kind::forward_scoring, in Compute()
|
| D | mkl_avgpooling_op.cc | 36 using mkldnn::prop_kind;
497 prop_kind pooling_prop_kind; in Compute()
501 pooling_prop_kind = prop_kind::forward_inference; in Compute()
503 pooling_prop_kind = prop_kind::forward_training; in Compute()
626 algorithm::pooling_avg_exclude_padding, prop_kind::forward_training); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 27 using mkldnn::prop_kind;
73 if ((context_.pkind == prop_kind::forward_training) || in Execute()
88 if ((context_.pkind == prop_kind::forward_training) || in Execute()
114 mkldnn::prop_kind pkind;
143 context_.pkind = fwdParams.training ? prop_kind::forward_training in Setup()
144 : prop_kind::forward_scoring; in Setup()
403 prop_kind::forward_training, src_md, bwdParams.eps, in Setup()
416 prop_kind::backward, diff_dst_md, src_md, bwdParams.eps, in Setup()
|
| D | mkl_maxpooling_op.cc | 35 using mkldnn::prop_kind;
574 prop_kind pooling_prop_kind; in Compute()
578 pooling_prop_kind = prop_kind::forward_inference; in Compute()
580 pooling_prop_kind = prop_kind::forward_training; in Compute()
736 prop_kind::forward_training); in Compute()
|
| D | mkl_lrn_op.cc | 43 using mkldnn::prop_kind;
143 auto lrn_desc = lrn_forward::desc(prop_kind::forward, lrn_across_channels, in Compute()
402 prop_kind::forward, lrn_across_channels, orig_input_md, kernel_size, in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 47 using mkldnn::prop_kind;
194 prop_kind::forward, convolution_direct, *context_.diff_src_md, in Setup()
|
| D | mkl_conv_grad_filter_ops.cc | 45 using mkldnn::prop_kind;
244 prop_kind::forward, convolution_direct, *context_.src_md, in Setup()
|
| D | mkl_relu_op.cc | 35 using mkldnn::prop_kind;
145 prop_kind::forward, fwdParams.alg_kind, *context_.src_md, in Setup()
352 prop_kind::forward_training, bwdParams.alg_kind, *context_.src_md, in Setup()
|
| D | mkl_conv_ops.h | 45 using mkldnn::prop_kind;
|
| D | mkl_conv_ops.cc | 51 using mkldnn::prop_kind;
232 prop_kind::forward, convolution_direct, *context_.src_md, in Setup()
238 prop_kind::forward, convolution_direct, *context_.src_md, in Setup()
|
| /external/v8/src/ast/ |
| D | prettyprinter.cc | 1056 const char* prop_kind = nullptr; in PrintClassProperties() local
1059 prop_kind = "METHOD"; in PrintClassProperties()
1062 prop_kind = "GETTER"; in PrintClassProperties()
1065 prop_kind = "SETTER"; in PrintClassProperties()
1068 prop_kind = "PUBLIC FIELD"; in PrintClassProperties()
1071 prop_kind = "PRIVATE FIELD"; in PrintClassProperties()
1076 prop_kind); in PrintClassProperties()
1135 const char* prop_kind = nullptr; in PrintObjectProperties() local
1138 prop_kind = "CONSTANT"; in PrintObjectProperties()
1141 prop_kind = "COMPUTED"; in PrintObjectProperties()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_mkl.cc | 47 using mkldnn::prop_kind;
110 prop_kind::forward_inference, convolution_direct, conv1_src_mem_desc, in MKLConvImpl()
|