| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_input_conversion_op.cc | 127 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
228 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_quantize_op.cc | 274 auto cpu_engine = engine(engine::kind::cpu, 0); in ComputeScalar() local
339 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_tfconv_op.h | 88 auto cpu_engine = engine(engine::kind::cpu, 0); in ConvertMklToTf() local
|
| H A D | mkl_transpose_op.cc | 68 engine cpu_engine = engine(engine::kind::cpu, 0); in MKLTransposeND() local
|
| H A D | mkl_dequantize_op.cc | 55 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_reshape_op.cc | 139 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_layer_norm_op.cc | 63 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_aggregate_ops.cc | 155 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_softmax_op.cc | 216 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| H A D | mkl_relu_op.cc | 551 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluOpBase
755 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluGradOpBase
|
| H A D | mkl_slice_op.cc | 376 auto cpu_engine = engine(engine::kind::cpu, 0); in ComputeMklSlice() local
|
| H A D | mkl_concat_op.cc | 468 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A 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
|
| H A D | mkl_util.h | 1737 MklPrimitive(const engine& cpu_engine) { cpu_engine_ = cpu_engine; } in MklPrimitive()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | LlgaTensorImpl.cpp | 36 static dnnl::engine cpu_engine = dnnl::graph::make_engine_with_allocator( in getEngine() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv2d_mkl.cc | 63 auto cpu_engine = engine(engine::cpu, 0); in MKLConvImpl() local
|