OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:filter_mkl_shape
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.cc
520
MklDnnShape src_mkl_shape,
filter_mkl_shape
;
variable
522
GetMklShape(context, kInputIndex_Filter, &
filter_mkl_shape
);
523
OP_REQUIRES(context,
filter_mkl_shape
.IsMklTensor() == false,
558
filter_mkl_shape
.SetMklTensor(false);
562
filter_mkl_shape
);
580
auto filter_md =
filter_mkl_shape
.IsMklTensor() // Should NEVER be true
581
?
filter_mkl_shape
.GetMklLayout()
698
MklDnnShape
filter_mkl_shape
;
local
699
filter_mkl_shape
.SetMklTensor(true);
700
filter_mkl_shape
.SetMklLayout(&filter_pd)
[
all
...]
mkl_conv_ops.h
368
MklDnnShape input_mkl_shape,
filter_mkl_shape
, outbprop_mkl_shape;
variable
370
GetMklShape(context, kFilterIdx, &
filter_mkl_shape
);
373
ValidateMklShapes(input_mkl_shape,
filter_mkl_shape
, outbprop_mkl_shape);
434
auto fwd_filter_md =
filter_mkl_shape
.IsMklTensor()
435
?
filter_mkl_shape
.GetMklLayout()
504
const MklDnnShape&
filter_mkl_shape
,
mkl_conv_grad_filter_ops.cc
448
const MklDnnShape&
filter_mkl_shape
,
450
CHECK(!
filter_mkl_shape
.IsMklTensor())
mkl_conv_grad_input_ops.cc
373
const MklDnnShape&
filter_mkl_shape
,
Completed in 81 milliseconds