OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:SetOpMemDesc
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc
90
src.
SetOpMemDesc
(src_dims, memory::format::nc);
mkl_conv_ops.h
481
input.
SetOpMemDesc
(fwd_input_dims, memory::format::any);
482
filter.
SetOpMemDesc
(fwd_filter_dims, memory::format::any);
483
outbackprop.
SetOpMemDesc
(outbprop_dims, memory::format::any);
484
output.
SetOpMemDesc
(bwd_output_dims, memory::format::any);
mkl_conv_ops.cc
593
src.
SetOpMemDesc
(src_dims, memory::format::any);
594
filter.
SetOpMemDesc
(filter_dims, memory::format::any);
595
output.
SetOpMemDesc
(output_dims_mkl_order, memory::format::any);
604
bias.
SetOpMemDesc
(bias_size, memory::format::any);
[
all
...]
mkl_lrn_op.cc
804
src_dnn_data.
SetOpMemDesc
(input_dims, memory::format::nhwc);
808
dst_dnn_data.
SetOpMemDesc
(input_dims, memory::format::nhwc);
[
all
...]
mkl_conv_grad_filter_ops.cc
518
bias_grad->
SetOpMemDesc
(bias_grad_dims, memory::format::x);
mkl_fused_batch_norm_op.cc
[
all
...]
mkl_avgpooling_op.cc
504
dnn_data_output.
SetOpMemDesc
(output_dims_mkl_order, memory::format::any);
mkl_maxpooling_op.cc
549
dnn_data_output.
SetOpMemDesc
(output_dims_mkl_order, memory::format::any);
[
all
...]
/external/tensorflow/tensorflow/core/util/
mkl_util.h
[
all
...]
Completed in 520 milliseconds