OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:original_input_dims_nchw
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h
332
memory::dims*
original_input_dims_nchw
, MklPoolParameters* pool_params,
334
CHECK_NOTNULL(
original_input_dims_nchw
);
339
*
original_input_dims_nchw
=
347
: memory::desc(*
original_input_dims_nchw
, MklDnnType<T>(),
mkl_avgpooling_op.cc
575
memory::dims output_dims_mkl_order,
original_input_dims_nchw
;
variable
579
&
original_input_dims_nchw
, &pool_params, &original_input_shape);
596
original_input_dims_nchw
,
627
context, pool_bkwd_prim_desc,
original_input_dims_nchw
,
Completed in 102 milliseconds