Home
last modified time | relevance | path

Searched defs:diff_src_data (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_grad_input_ops.cc93 void Execute(const T* diff_src_data, const T* filter_data, in Execute()
360 auto diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local
457 T* diff_src_data = in Compute() local
H A Dmkl_fused_batch_norm_op.cc438 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute()
1357 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); in Compute() local
1418 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); in HandleEmptyInput() local
H A Dmkl_pooling_ops_common.cc190 T* diff_src_data, const void* ws_data, in Execute()
H A Dmkl_avgpooling_op.cc288 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
H A Dmkl_maxpooling_op.cc340 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
H A Dmkl_relu_op.cc260 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data, in Execute()
739 T* diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local