| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_conv_grad_filter_ops.cc | 106 context_.diff_dst_mem->set_data_handle(
114 context_.diff_dst_mem->set_data_handle(DummyData);
128 context_.diff_dst_mem->set_data_handle(
135 context_.diff_dst_mem->set_data_handle(DummyData);
172 std::shared_ptr<mkldnn::memory> diff_dst_mem; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
196 diff_dst_mem(nullptr),
271 context_.diff_dst_mem.reset(
283 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem,
287 *context_.bwd_filter_pd, *context_.src_mem, *context_.diff_dst_mem,
|
| mkl_conv_grad_input_ops.cc | 104 context_.diff_dst_mem->set_data_handle(
112 context_.diff_dst_mem->set_data_handle(DummyData);
137 std::shared_ptr<mkldnn::memory> diff_dst_mem; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
163 diff_dst_mem(nullptr),
211 context_.diff_dst_mem.reset(new memory(
228 *context_.bwd_input_pd, *context_.diff_dst_mem, *context_.filter_mem,
|
| mkl_pooling_ops_common.h | 265 std::shared_ptr<mkldnn::memory> diff_dst_mem; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
289 diff_dst_mem(nullptr),
|
| mkl_relu_op.cc | 267 context_.diff_dst_mem->set_data_handle(
274 context_.diff_dst_mem->set_data_handle(DummyData);
295 std::shared_ptr<memory> diff_dst_mem; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
325 diff_dst_mem(nullptr),
364 context_.diff_dst_mem.reset(new memory(*context_.diff_dst_mpd, DummyData));
370 *context_.bwd_pd, *context_.src_mem, *context_.diff_dst_mem,
[all...] |
| mkl_fused_batch_norm_op.cc | 315 context_.diff_dst_mem->set_data_handle(
333 context_.diff_dst_mem->set_data_handle(DummyData);
346 return (*context_.diff_dst_mem).get_primitive_desc().desc().data.format;
362 std::shared_ptr<mkldnn::memory> diff_dst_mem; member in struct:tensorflow::MklFusedBatchNormBwdPrimitive::BatchNormBwdContext
376 diff_dst_mem(nullptr),
424 context_.diff_dst_mem.reset(
437 *context_.variance_mem, *context_.diff_dst_mem, *context_.weights_mem,
[all...] |
