HomeSort by relevance Sort by last modified time
    Searched refs:diff_dst_dims (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 387 memory::dims diff_dst_dims = local
393 memory::desc(diff_dst_dims, MklDnnType<T>(), original_output_format);
mkl_relu_op.cc 528 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), local
531 memory::desc(diff_dst_dims, MklDnnType<T>(), src_mkl_data_format);
    [all...]
mkl_fused_batch_norm_op.cc 1134 memory::dims src_dims, diff_dst_dims; variable
    [all...]

Completed in 182 milliseconds