Home
last modified time | relevance | path

Searched defs:cpu_engine_ (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_avgpooling_op.cc176 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingOp
307 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingGradOp
H A Dmkl_maxpooling_op.cc206 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingOp
363 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingGradOp
H A Dmkl_lrn_op.cc310 engine cpu_engine_; member in tensorflow::MklLRNOp
657 engine cpu_engine_; member in tensorflow::MklLRNGradOp
H A Dmkl_requantize_per_channel_op.cc178 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklRequantizePerChannelOp
H A Dmkl_fused_batch_norm_op.cc988 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklFusedBatchNormOp
1402 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklFusedBatchNormGradOp
H A Dmkl_conv_grad_input_ops.cc511 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvCustomBackpropInputOp
H A Dmkl_conv_grad_filter_ops.cc649 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvCustomBackpropFilterOp
H A Dmkl_matmul_ops_common.h563 engine cpu_engine_ = engine(engine::kind::cpu, 0); variable
H A Dmkl_conv_ops.cc1169 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvOp
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h1301 const engine* cpu_engine_; variable
1740 engine cpu_engine_ = engine(engine::kind::cpu, 0); variable