Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_requantize_per_channel_op.cc162 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklRequantizePerChannelOp
Dmkl_avgpooling_op.cc565 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklAvgPoolingOp
670 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklAvgPoolingGradOp
Dmkl_conv_grad_filter_ops.cc295 engine cpu_engine_; member in tensorflow::MklConvBwdFilterPrimitive
608 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklConvCustomBackpropFilterOp
Dmkl_slice_op.cc217 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklSlicePrimitive
Dmkl_fused_batch_norm_op.cc219 engine cpu_engine_; member in tensorflow::MklFusedBatchNormFwdPrimitive
443 engine cpu_engine_; member in tensorflow::MklFusedBatchNormBwdPrimitive
Dmkl_conv_grad_input_ops.cc235 engine cpu_engine_; member in tensorflow::MklConvBwdInputPrimitive
Dmkl_relu_op.cc163 engine cpu_engine_; member in tensorflow::MklEltwiseFwdPrimitive
377 engine cpu_engine_; member in tensorflow::MklEltwiseBwdPrimitive
Dmkl_conv_ops.cc316 engine cpu_engine_; member in tensorflow::MklConvFwdPrimitive
1279 engine cpu_engine_ = engine(engine::cpu, 0); member in tensorflow::MklConvOp
Dmkl_pooling_ops_common.h149 engine cpu_engine_; variable
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1664 const engine* cpu_engine_; variable
2281 engine cpu_engine_ = engine(engine::cpu, 0); variable