| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 140 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
238 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_tfconv_op.h | 85 auto cpu_engine = engine(engine::cpu, 0); in ConvertMklToTf() local
|
| D | mkl_softmax_op.cc | 48 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_transpose_op.cc | 127 engine cpu_engine = engine(engine::cpu, 0); in MKLTransposeND() local
|
| D | mkl_reshape_op.cc | 129 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_aggregate_ops.cc | 50 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_lrn_op.cc | 94 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
343 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_maxpooling_op.cc | 652 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklMaxPoolingOp
688 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_concat_op.cc | 157 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
|
| D | mkl_slice_op.cc | 361 auto cpu_engine = engine(engine::cpu, 0); in ComputeMklSlice() local
|
| D | mkl_fused_batch_norm_op.cc | 723 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklFusedBatchNormOp
1045 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklFusedBatchNormGradOp
|
| D | mkl_conv_grad_input_ops.cc | 480 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklConvCustomBackpropInputOp
|
| D | mkl_relu_op.cc | 543 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklReluOpBase
711 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklReluGradOpBase
|
| D | mkl_pooling_ops_common.h | 301 engine cpu_engine; variable
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 28 auto cpu_engine = engine(engine::cpu, 0); in TEST() local
60 auto cpu_engine = engine(engine::cpu, 0); in TEST() local
|
| D | mkl_util.h | 845 auto cpu_engine = engine(engine::cpu, 0); in ConvertMklToTF() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_mkl.cc | 61 auto cpu_engine = engine(engine::cpu, 0); in MKLConvImpl() local
|