OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:dnn_shape_diff_dst
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc
487
MklDnnShape dnn_shape_src,
dnn_shape_diff_dst
;
local
489
GetMklShape(context, diff_dst_index, &
dnn_shape_diff_dst
);
510
if (!dnn_shape_src.IsMklTensor() && !
dnn_shape_diff_dst
.IsMklTensor()) {
518
!
dnn_shape_diff_dst
.IsMklTensor()) {
533
dnn_shape_diff_dst
.IsMklTensor()) {
535
diff_dst_md =
dnn_shape_diff_dst
.GetMklLayout();
538
dnn_shape_diff_dst
.GetTfDataFormat();
549
diff_dst_md =
dnn_shape_diff_dst
.GetMklLayout();
560
if (dnn_shape_src.IsMklTensor() ||
dnn_shape_diff_dst
.IsMklTensor()) {
676
MklDnnShape
dnn_shape_diff_dst
;
local
743
MklDnnShape
dnn_shape_diff_dst
;
local
814
MklDnnShape
dnn_shape_diff_dst
;
local
[
all
...]
mkl_fused_batch_norm_op.cc
1061
MklDnnShape dnn_shape_src,
dnn_shape_diff_dst
;
variable
[
all
...]
Completed in 93 milliseconds