Searched defs:cpu_engine (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 139 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
240 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_tfconv_op.h | 88 auto cpu_engine = engine(engine::kind::cpu, 0); in ConvertMklToTf() local
|
| D | mkl_dequantize_op.cc | 53 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_transpose_op.cc | 68 engine cpu_engine = engine(engine::kind::cpu, 0); in MKLTransposeND() local
|
| D | mkl_reshape_op.cc | 140 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_quantize_op.cc | 265 auto cpu_engine = engine(engine::kind::cpu, 0); in ComputeScalar() local
330 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_aggregate_ops.cc | 156 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_softmax_op.cc | 202 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| D | mkl_slice_op.cc | 369 auto cpu_engine = engine(engine::kind::cpu, 0); in ComputeMklSlice() local
|
| D | mkl_relu_op.cc | 550 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluOpBase
754 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluGradOpBase
|
| D | mkl_concat_op.cc | 458 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 25 auto cpu_engine = engine(engine::kind::cpu, 0); in TEST() local
59 auto cpu_engine = engine(engine::kind::cpu, 0); in TEST() local
|
| D | mkl_util.h | 1696 MklPrimitive(const engine& cpu_engine) { cpu_engine_ = cpu_engine; } in MklPrimitive()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_mkl.cc | 61 auto cpu_engine = engine(engine::cpu, 0); in MKLConvImpl() local
|