HomeSort by relevance Sort by last modified time
    Searched defs:diff_dst_fmt (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_conv_grad_filter_ops.cc 142 return context_.diff_dst_fmt;
160 memory::format diff_dst_fmt; member in struct:tensorflow::MklConvBwdFilterPrimitive::ConvBwdFilterContext
191 diff_dst_fmt(memory::format::any),
263 context_.diff_dst_fmt = static_cast<mkldnn::memory::format>(
mkl_conv_grad_input_ops.cc 119 return context_.diff_dst_fmt;
132 memory::format diff_dst_fmt; member in struct:tensorflow::MklConvBwdInputPrimitive::ConvBwdInputContext
160 diff_dst_fmt(memory::format::any),
220 context_.diff_dst_fmt =
mkl_pooling_ops_common.h 238 memory::format GetDiffDstFormat() const { return context_.diff_dst_fmt; }
255 mkldnn::memory::format diff_dst_fmt; member in struct:tensorflow::MklPoolingBwdPrimitive::PoolingBwdContext
285 diff_dst_fmt(memory::format::any),
mkl_relu_op.cc 249 context_.diff_dst_fmt =
284 memory::format GetDiffDstMemoryFormat() { return context_.diff_dst_fmt; }
291 memory::format diff_dst_fmt; member in struct:tensorflow::MklEltwiseBwdPrimitive::EltwiseBwdContext
323 diff_dst_fmt(memory::format::any),
393 auto diff_dst_fmt = local
399 bwdParams, src_fmt, diff_dst_fmt));
404 bwdParams, src_fmt, diff_dst_fmt, eltwise_backward);
417 const memory::format& diff_dst_fmt) {
426 key_creator.AddAsKey(static_cast<int>(diff_dst_fmt));
432 const memory::format& diff_dst_fmt) {
    [all...]

Completed in 3368 milliseconds