OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:input_mkl_shape
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h
195
const MklDnnShape&
input_mkl_shape
,
203
input_mkl_shape
.IsMklTensor()
204
?
input_mkl_shape
.GetMklLayout()
209
this->InitMklPoolParameters(context, pool_params,
input_mkl_shape
,
254
const MklDnnShape&
input_mkl_shape
) {
255
if (!
input_mkl_shape
.IsMklTensor()) {
259
OP_REQUIRES(context,
input_mkl_shape
.GetDimension() == 4,
mkl_conv_ops.h
368
MklDnnShape
input_mkl_shape
, filter_mkl_shape, outbprop_mkl_shape;
variable
369
GetMklShape(context, kInputIdx, &
input_mkl_shape
);
373
ValidateMklShapes(
input_mkl_shape
, filter_mkl_shape, outbprop_mkl_shape);
429
input_mkl_shape
.IsMklTensor()
430
?
input_mkl_shape
.GetMklLayout()
503
virtual void ValidateMklShapes(const MklDnnShape&
input_mkl_shape
,
mkl_conv_grad_input_ops.cc
372
void ValidateMklShapes(const MklDnnShape&
input_mkl_shape
,
378
CHECK(!
input_mkl_shape
.IsMklTensor())
mkl_conv_grad_filter_ops.cc
447
void ValidateMklShapes(const MklDnnShape&
input_mkl_shape
,
/external/tensorflow/tensorflow/core/util/
mkl_util.h
791
MklDnnShape
input_mkl_shape
;
local
792
GetMklShape(context, input_idx, &
input_mkl_shape
);
793
if (
input_mkl_shape
.IsMklTensor()) {
794
return
input_mkl_shape
.GetTfShape();
[
all
...]
Completed in 704 milliseconds