Home
last modified time | relevance | path

Searched refs:diff_dst_tensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc595 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute() local
638 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
699 const T* diff_dst_data = diff_dst_tensor.flat<T>().data(); in Compute()
701 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
806 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute_Scalar() local
815 diff_dst_tensor.shape(), dnn_shape_diff_src); in Compute_Scalar()
820 static_cast<void*>(const_cast<T*>(diff_dst_tensor.flat<T>().data())); in Compute_Scalar()
878 const Tensor& diff_dst_tensor = MklGetInput(context, diff_dst_index); in Compute_Scalar() local
887 diff_dst_tensor.shape(), dnn_shape_diff_src); in Compute_Scalar()
892 static_cast<void*>(const_cast<T*>(diff_dst_tensor.flat<T>().data())); in Compute_Scalar()
[all …]
Dmkl_conv_grad_input_ops.cc317 const Tensor& diff_dst_tensor = MklGetInput(context, kOutbpropIdx); in Compute() local
335 diff_dst_tensor.shape(), this->data_format_, in Compute()
475 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
481 static_cast<T*>(const_cast<T*>(diff_dst_tensor.flat<T>().data())); in Compute()
Dmkl_fused_batch_norm_op.cc1070 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIndex); in Compute() local
1090 diff_dst_tensor.shape().DebugString())); in Compute()
1092 tf_shape_diff_dst = diff_dst_tensor.shape(); in Compute()
1094 context, diff_dst_tensor.dims() == 4, in Compute()
1096 diff_dst_tensor.shape().DebugString())); in Compute()
1168 : TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
1184 static_cast<T*>(const_cast<T*>(diff_dst_tensor.flat<T>().data())); in Compute()
1194 reorder_diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
Dmkl_conv_grad_filter_ops.cc374 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIdx); in Compute() local
553 diff_dst.SetUsrMem(diff_dst_md, &diff_dst_tensor); in Compute()
559 static_cast<T*>(const_cast<T*>(diff_dst_tensor.flat<T>().data())); in Compute()