Home
last modified time | relevance | path

Searched refs:execute_primitives (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op.cc157 execute_primitives(net, reorder_stream, reorder_net_args); in Compute()
Dmkl_pooling_ops_common.cc110 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
210 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
Dmkl_transpose_op.cc154 execute_primitives(net, transpose_stream, net_args); in MKLTransposeND()
Dmkl_softmax_op.cc73 execute_primitives(context_.fwd_primitives, fwd_cpu_stream, in Execute()
Dmkl_matmul_ops_common.h123 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
555 execute_primitives(context_.matmul_primitives, stream, context_.net_args); in Execute()
Dmkl_slice_op.cc198 execute_primitives(context_.slice_primitives, slice_stream, in Execute()
Dmkl_conv_grad_input_ops.cc125 execute_primitives(context_.bwd_input_primitives, bwd_input_stream, in Execute()
Dmkl_relu_op.cc89 execute_primitives(context_.fwd_primitives, fwd_stream, in Execute()
274 execute_primitives(context_.bwd_primitives, bwd_stream, in Execute()
Dmkl_conv_grad_filter_ops.cc133 execute_primitives(context_.bwd_filter_primitives, bwd_filter_stream, in Execute()
Dmkl_fused_batch_norm_op.cc122 execute_primitives(context_.fwd_primitives, fwd_stream, context_.net_args); in Execute()
466 execute_primitives(context_.bwd_primitives, bwd_stream, context_.net_args); in Execute()
Dmkl_concat_op.cc301 execute_primitives(context_.fwd_primitives, fwd_stream,
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h130 inline void execute_primitives( in execute_primitives() function
1504 execute_primitives(net, cpu_stream, net_args);
1565 execute_primitives(net, cpu_stream, net_args);
1666 execute_primitives(net, cpu_stream, net_args);