HomeSort by relevance Sort by last modified time
    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