experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Calling refactored method

Browse Source
This commit is contained in:
mbhuiyan 2020-02-13 15:02:33 -08:00
parent b6407acdd5
commit aba36d0286
2 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc
View File

@ -115,13 +115,8 @@ class MklConvBwdFilterPrimitive : public MklPrimitive {
static_cast<void*>(const_cast<T*>(diff_dst_data)));
#ifdef ENABLE_MKLDNN_V1
DCHECK_EQ(context_.bwd_filter_primitives.size(),
context_.bwd_filter_primitives_args.size());
for (size_t i = 0; i < context_.bwd_filter_primitives.size(); ++i) {
context_.bwd_filter_primitives.at(i).execute(
*context_.bwd_filter_stream,
context_.bwd_filter_primitives_args.at(i));
}
execute_primitives(context_.bwd_filter_primitives,
context_.bwd_filter_stream, context_.bwd_filter_primitives_args);
#else
context_.bwd_filter_stream->submit(context_.bwd_filter_primitives);
#endif

8
tensorflow/core/kernels/mkl_conv_grad_input_ops.cc
View File

@ -112,12 +112,8 @@ class MklConvBwdInputPrimitive : public MklPrimitive {
static_cast<T*>(const_cast<T*>(diff_dst_data)));
#ifdef ENABLE_MKLDNN_V1
DCHECK_EQ(context_.bwd_input_primitives.size(),
context_.bwd_input_primitives_args.size());
for (size_t i = 0; i < context_.bwd_input_primitives.size(); ++i) {
context_.bwd_input_primitives.at(i).execute(
*context_.bwd_input_stream, context_.bwd_input_primitives_args.at(i));
}
execute_primitives(context_.bwd_input_primitives, context_.bwd_input_stream,
context_.bwd_input_primitives_args);
#else
context_.bwd_input_stream->submit(context_.bwd_input_primitives);
#endif // ENABLE_MKLDNN_V1