Home
last modified time | relevance | path

Searched refs:cpu_engine (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc28 auto cpu_engine = engine(engine::cpu, 0); in TEST() local
29 MklDnnData<float> a(&cpu_engine); in TEST()
60 auto cpu_engine = engine(engine::cpu, 0); in TEST() local
63 MklDnnData<float> a(&cpu_engine); in TEST()
75 MklDnnData<float> b(&cpu_engine); in TEST()
Dmkl_util.h845 auto cpu_engine = engine(engine::cpu, 0); in ConvertMklToTF() local
846 MklDnnData<T> input(&cpu_engine); in ConvertMklToTF()
851 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTF()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_input_conversion_op.cc140 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
141 MklDnnData<T> input(&cpu_engine); in Compute()
147 memory::primitive_desc(input1_md, cpu_engine), in Compute()
238 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
239 MklDnnData<T> tf_input(&cpu_engine); in Compute()
246 memory::primitive_desc(output_mkl_md, cpu_engine), tensor_out, &net); in Compute()
Dmkl_lrn_op.cc94 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
120 MklDnnData<T> src_dnn_data(&cpu_engine); in Compute()
121 MklDnnData<T> dst_dnn_data(&cpu_engine); in Compute()
122 MklDnnData<uint8> workspace_dnn_data(&cpu_engine); in Compute()
146 auto lrn_prim_desc = lrn_forward::primitive_desc(lrn_desc, cpu_engine); in Compute()
343 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
344 MklDnnData<T> input_grad_dnn_data(&cpu_engine); in Compute()
345 MklDnnData<T> orig_input_dnn_data(&cpu_engine); in Compute()
346 MklDnnData<T> orig_output_dnn_data(&cpu_engine); in Compute()
347 MklDnnData<T> output_dnn_data(&cpu_engine); in Compute()
[all …]
Dmkl_aggregate_ops.cc50 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
105 MklDnnData<T> src1(&cpu_engine); in Compute()
106 MklDnnData<T> src2(&cpu_engine); in Compute()
107 MklDnnData<T> dst(&cpu_engine); in Compute()
194 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine)); in Compute()
196 srcs_pd.push_back(memory::primitive_desc(common_md, cpu_engine)); in Compute()
Dmkl_softmax_op.cc48 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
114 MklDnnData<T> src(&cpu_engine); in Compute()
115 MklDnnData<T> dst(&cpu_engine); in Compute()
133 softmax_forward::primitive_desc(softmax_fwd_desc, cpu_engine); in Compute()
Dmkl_tfconv_op.h85 auto cpu_engine = engine(engine::cpu, 0); in ConvertMklToTf() local
86 MklDnnData<T> input(&cpu_engine); in ConvertMklToTf()
93 auto output_tf_pd = memory::primitive_desc(output_tf_md, cpu_engine); in ConvertMklToTf()
Dmkl_reshape_op.cc129 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
130 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
152 memory::primitive_desc(output_tf_md, cpu_engine); in Compute()
183 memory::primitive_desc(output_tf_md, cpu_engine); in Compute()
Dmkl_maxpooling_op.cc521 MklDnnData<T> dnn_data_input(&cpu_engine); in Compute()
522 MklDnnData<T> dnn_data_output(&cpu_engine); in Compute()
601 cpu_engine); in Compute()
631 MklDnnData<uint8> dnn_data_wksp(&cpu_engine); in Compute()
652 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklMaxPoolingOp
688 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
700 MklDnnData<T> grad_dnn_data(&cpu_engine); in Compute()
701 MklDnnData<uint8> workspace_dnn_data(&cpu_engine); in Compute()
756 cpu_engine); in Compute()
775 cpu_engine); in Compute()
Dmkl_transpose_op.cc127 engine cpu_engine = engine(engine::cpu, 0); in MKLTransposeND() local
128 MklDnnData<T> in(&cpu_engine); in MKLTransposeND()
129 MklDnnData<T> out(&cpu_engine); in MKLTransposeND()
Dmkl_fused_batch_norm_op.cc595 MklDnnData<T> src(&cpu_engine); in Compute()
596 MklDnnData<T> weights(&cpu_engine); in Compute()
652 cpu_engine); in Compute()
723 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklFusedBatchNormOp
919 MklDnnData<T> src(&cpu_engine); in Compute()
920 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
921 MklDnnData<T> weights(&cpu_engine); in Compute()
922 MklDnnData<T> diff_weights(&cpu_engine); in Compute()
969 cpu_engine); in Compute()
982 cpu_engine); in Compute()
[all …]
Dmkl_slice_op.cc361 auto cpu_engine = engine(engine::cpu, 0); in ComputeMklSlice() local
362 MklDnnData<T> src(&cpu_engine); in ComputeMklSlice()
363 MklDnnData<T> output(&cpu_engine); in ComputeMklSlice()
418 auto output_pd = memory::primitive_desc(output_md, cpu_engine); in ComputeMklSlice()
Dmkl_pooling_ops_common.cc162 new pooling_forward::primitive_desc(*context_.fwd_desc, cpu_engine)); in Setup()
164 *context_.bwd_desc, cpu_engine, *context_.fwd_pd)); in Setup()
176 cpu_engine}, in Setup()
186 {{{context_.ws_dims}, context_.ws_dt, context_.ws_fmt}, cpu_engine}, in Setup()
Dmkl_concat_op.cc157 auto cpu_engine = engine(engine::cpu, 0); in Compute() local
290 std::vector<MklDnnData<T>> srcs(N, MklDnnData<T>(&cpu_engine)); in Compute()
326 srcs_pd.push_back(memory::primitive_desc(src_md, cpu_engine)); in Compute()
348 MklDnnData<T> dst(&cpu_engine); in Compute()
Dmkl_relu_op.cc470 MklDnnData<T> src(&cpu_engine); in Compute()
496 cpu_engine); in Compute()
543 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklReluOpBase
564 MklDnnData<T> src(&cpu_engine); in Compute()
565 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
711 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklReluGradOpBase
Dmkl_conv_grad_input_ops.cc309 MklDnnData<T> filter(&cpu_engine); in Compute()
310 MklDnnData<T> diff_dst(&cpu_engine); in Compute()
480 engine cpu_engine = engine(engine::cpu, 0); member in tensorflow::MklConvCustomBackpropInputOp
Dmkl_pooling_ops_common.h214 : cpu_engine(engine::cpu, 0) { in MklPoolingBwdPrimitive()
301 engine cpu_engine; variable
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_mkl.cc61 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()