Home
last modified time | relevance | path

Searched defs:diff_dst_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_pooling_ops_common.cc188 void MklPoolingBwdPrimitive<T>::Execute(const T* diff_dst_data, in Execute()
Dmkl_conv_grad_filter_ops.cc105 const T* diff_bias_data, const T* diff_dst_data, in Execute()
149 const T* diff_dst_data, in Execute()
550 T* diff_dst_data = nullptr; in Compute() local
Dmkl_conv_grad_input_ops.cc107 const T* diff_dst_data, in Execute()
472 T* diff_dst_data = nullptr; in Compute() local
Dmkl_avgpooling_op.cc264 T* diff_dst_data = nullptr; in Compute() local
Dmkl_maxpooling_op.cc317 T* diff_dst_data = nullptr; in Compute() local
Dmkl_relu_op.cc256 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data, in Execute()
699 const T* diff_dst_data = diff_dst_tensor.flat<T>().data(); in Compute() local
Dmkl_fused_batch_norm_op.cc422 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute()
1183 T* diff_dst_data = in Compute() local