OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:dnn_shape_diff_src
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc
580
MklDnnShape
dnn_shape_diff_src
;
local
583
dnn_shape_diff_src
.SetMklTensor(true);
585
dnn_shape_diff_src
.SetMklLayout(&diff_src_pd);
586
dnn_shape_diff_src
.SetElemType(MklDnnType<T>());
587
dnn_shape_diff_src
.SetTfLayout(dnn_shape_src.GetDimension(),
592
dnn_shape_diff_src
.SetMklTensor(false);
596
tf_shape_diff_src,
dnn_shape_diff_src
);
679
MklDnnShape
dnn_shape_diff_src
;
local
680
dnn_shape_diff_src
.SetMklTensor(false);
682
diff_dst_tensor.shape(),
dnn_shape_diff_src
);
746
MklDnnShape
dnn_shape_diff_src
;
local
817
MklDnnShape
dnn_shape_diff_src
;
local
[
all
...]
mkl_fused_batch_norm_op.cc
1212
MklDnnShape
dnn_shape_diff_src
;
variable
1296
MklDnnShape
dnn_shape_diff_src
;
local
[
all
...]
Completed in 82 milliseconds