| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_bias_ops.cc | 189 CHECK_EQ(dnnConversionCreate_F32(&convert_outbackprop,
219 CHECK_EQ(dnnConversionCreate_F32(&mkl_convert_input, lt_input,
|
| mkl_conv_grad_filter_ops.cc | 347 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input, lt_input,
373 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_out_backprop,
406 CHECK_EQ(dnnConversionCreate_F32(&convert_bwdfilter,
|
| mkl_aggregate_ops.cc | 228 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input1, lt_input1,
253 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input2, lt_input2,
|
| mkl_conv_grad_input_ops.cc | 295 CHECK_EQ(dnnConversionCreate_F32(&mkl_convert_filter, lt_filter,
323 dnnConversionCreate_F32(&mkl_convert_outbackprop, lt_outbackprop,
|
| mkl_conv_ops.cc | 404 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input, lt_input,
428 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_filter, lt_filter,
455 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_bias, lt_bias,
[all...] |
| mkl_avgpooling_op.cc | 187 CHECK_EQ(dnnConversionCreate_F32(&convert_input, lt_user_input,
278 CHECK_EQ(dnnConversionCreate_F32(&mkl_context.convert_outbackprop,
|
| mkl_maxpooling_op.cc | 369 CHECK_EQ(dnnConversionCreate_F32(&convert_input, lt_input_user,
378 dnnConversionCreate_F32(&convert_outbackprop, lt_outbackprop_user,
[all...] |
| mkl_fused_batch_norm_op.cc | 294 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input, mkl_lt_input,
590 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input, mkl_lt_input,
619 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_out_backprop,
[all...] |
| mkl_lrn_op.cc | 198 CHECK_EQ(dnnConversionCreate_F32(&convert_input, lt_input,
575 dnnConversionCreate_F32(&convert_input, lt_input, lt_bdw_input),
605 CHECK_EQ(dnnConversionCreate_F32(
[all...] |
| mkl_concat_op.cc | 456 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input,
[all...] |
| mkl_relu_op.cc | 216 CHECK_EQ(dnnConversionCreate_F32(&cv_input_to_grad, lt_input, lt_grad),
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 207 CHECK_EQ(dnnConversionCreate_F32(&convert, curLayout, targetLayout),
[all...] |
