Searched defs:cpu_engine_ (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 166 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingOp
296 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklAvgPoolingGradOp
|
| D | mkl_maxpooling_op.cc | 200 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingOp
356 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklMaxPoolingGradOp
|
| D | mkl_requantize_per_channel_op.cc | 168 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklRequantizePerChannelOp
|
| D | mkl_lrn_op.cc | 309 engine cpu_engine_; member in tensorflow::MklLRNOp
656 engine cpu_engine_; member in tensorflow::MklLRNGradOp
|
| D | mkl_conv_grad_input_ops.cc | 508 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvCustomBackpropInputOp
|
| D | mkl_fused_batch_norm_op.cc | 926 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklFusedBatchNormOp
1309 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklFusedBatchNormGradOp
|
| D | mkl_conv_grad_filter_ops.cc | 620 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvCustomBackpropFilterOp
|
| D | mkl_matmul_ops_common.h | 491 engine cpu_engine_ = engine(engine::kind::cpu, 0); variable
|
| D | mkl_conv_ops.cc | 933 engine cpu_engine_ = engine(engine::kind::cpu, 0); member in tensorflow::MklConvOp
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1264 const engine* cpu_engine_; variable
1678 engine cpu_engine_ = engine(engine::kind::cpu, 0); variable
|