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

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 484 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); local
528 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(),
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor);
673 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); local
682 diff_dst_tensor.shape(), dnn_shape_diff_src);
687 static_cast<void*>(const_cast<T*>(diff_dst_tensor.flat<T>().data()));
740 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); local
749 diff_dst_tensor.shape(), dnn_shape_diff_src);
754 static_cast<void*>(const_cast<T*>(diff_dst_tensor.flat<T>().data()));
811 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index) local
    [all...]
mkl_fused_batch_norm_op.cc 1054 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIndex); variable
    [all...]

Completed in 624 milliseconds