OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:fwd_input_dims
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
mkl_conv_ops.h
408
memory::dims outbprop_dims,
fwd_input_dims
, fwd_filter_dims;
variable
415
input_tf_shape, filter_tf_shape, &
fwd_input_dims
, &fwd_filter_dims,
431
: memory::desc(
fwd_input_dims
, MklDnnType<T>(), tf_fmt);
476
auto bwd_output_dims = GetOutputDims(
fwd_input_dims
, fwd_filter_dims);
481
input.SetOpMemDesc(
fwd_input_dims
, memory::format::any);
527
/// input shape (
fwd_input_dims
) and filter shape (fwd_filter_dims).
529
const memory::dims&
fwd_input_dims
,
mkl_conv_grad_input_ops.cc
407
const memory::dims& GetOutputDims(const memory::dims&
fwd_input_dims
,
410
return
fwd_input_dims
;
mkl_conv_grad_filter_ops.cc
480
const memory::dims& GetOutputDims(const memory::dims&
fwd_input_dims
,
Completed in 1205 milliseconds