| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 139 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
140 MklDnnData<T> input(&cpu_engine); in Compute()
150 cpu_engine), in Compute()
153 ExecutePrimitive(net, &net_args, cpu_engine, context); in Compute()
240 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
241 MklDnnData<T> tf_input(&cpu_engine); in Compute()
248 net, net_args, cpu_engine); in Compute()
260 ExecutePrimitive(net, &net_args, cpu_engine, context); in Compute()
|
| D | mkl_dequantize_op.cc | 53 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
74 MklDnnData<T> src(&cpu_engine); in Compute()
75 MklDnnData<float> dst(&cpu_engine); in Compute()
78 reorder_stream.reset(CreateStream(ctx, cpu_engine)); in Compute()
151 ReorderPd(cpu_engine, src.GetUsrMem()->get_desc(), cpu_engine, in Compute()
|
| D | mkl_aggregate_ops.cc | 156 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
161 MklDnnData<T> dst(&cpu_engine); in Compute()
162 MklDnnData<T> src(&cpu_engine); in Compute()
178 fwd_cpu_stream.reset(CreateStream(ctx, cpu_engine)); in Compute()
217 auto sum_pd = sum::primitive_desc(coeff, srcs_pd, cpu_engine); in Compute()
|
| D | mkl_tfconv_op.h | 88 auto cpu_engine = engine(engine::kind::cpu, 0); in ConvertMklToTf() local
89 MklDnnData<T> input(&cpu_engine); in ConvertMklToTf()
|
| D | mkl_transpose_op.cc | 128 engine cpu_engine = engine(engine::kind::cpu, 0); in MKLTransposeND() local
129 MklDnnData<T> in(&cpu_engine); in MKLTransposeND()
130 MklDnnData<T> out(&cpu_engine); in MKLTransposeND()
|
| D | mkl_reshape_op.cc | 140 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
141 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
|
| 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
368 MklDnnData<float> src(&cpu_engine); in Compute()
369 MklDnnData<T> dst(&cpu_engine); in Compute()
|
| D | mkl_slice_op.cc | 369 auto cpu_engine = engine(engine::kind::cpu, 0); in ComputeMklSlice() local
370 MklDnnData<T> src(&cpu_engine); in ComputeMklSlice()
371 MklDnnData<T> output(&cpu_engine); in ComputeMklSlice()
436 src.CheckReorderToOpMem(op_md, cpu_engine, context); in ComputeMklSlice()
|
| D | mkl_relu_op.cc | 462 MklDnnData<T> src(&cpu_engine); in Compute()
488 src.CheckReorderToOpMem(eltwise_fwd_pd->src_desc(), cpu_engine, in Compute()
549 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluOpBase
587 MklDnnData<T> src(&cpu_engine); in Compute()
588 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
694 cpu_engine, context); in Compute()
703 cpu_engine, context); in Compute()
752 engine cpu_engine = engine(engine::kind::cpu, 0); member in tensorflow::MklReluGradOpBase
|
| D | mkl_concat_op.cc | 456 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
591 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine)); in Compute()
663 MklDnnData<T> dst(&cpu_engine); in Compute()
699 srcs[k].CheckReorderToOpMem(srcs_pd[k], cpu_engine, context); in Compute()
718 concat::primitive_desc(concat_dim, srcs_pd, cpu_engine); in Compute()
735 fwd_cpu_stream.reset(CreateStream(context, cpu_engine)); in Compute()
|
| D | mkl_softmax_op.cc | 202 auto cpu_engine = engine(engine::kind::cpu, 0); in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 27 auto cpu_engine = engine(ENGINE_CPU, 0); in TEST() local
28 MklDnnData<float> a(&cpu_engine); in TEST()
59 auto cpu_engine = engine(ENGINE_CPU, 0); in TEST() local
62 MklDnnData<float> a(&cpu_engine); in TEST()
73 MklDnnData<float> b(&cpu_engine); in TEST()
|
| D | mkl_types.h | 74 #define MEMORY_CONSTRUCTOR_WITH_MEM_PD(mem_ptr, cpu_engine, data) \ argument
75 memory(GET_MEMORY_DESC_FROM_MEM_PTR(mem_ptr), cpu_engine, data)
|
| D | mkl_util.h | 611 const engine& cpu_engine,
615 stream* cpu_stream = CreateStream(context, cpu_engine);
639 engine cpu_engine(engine::kind::cpu, 0); in ConvertMklToTF()
640 MklDnnData<T> input(&cpu_engine); in ConvertMklToTF()
651 net, net_args, cpu_engine); in ConvertMklToTF()
656 ExecutePrimitive(net, &net_args, cpu_engine, context); in ConvertMklToTF()
1675 MklPrimitive(const engine& cpu_engine) { cpu_engine_ = cpu_engine; } in MklPrimitive() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_mkl.cc | 61 auto cpu_engine = engine(engine::cpu, 0); in MKLConvImpl() local
88 cpu_engine}, in MKLConvImpl()
92 cpu_engine}, in MKLConvImpl()
96 cpu_engine}, in MKLConvImpl()
114 convolution_forward::primitive_desc(conv1_desc, cpu_engine); in MKLConvImpl()
|