Searched defs:diff_src_data (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 106 void Execute(const T* diff_src_data, const T* filter_data, in Execute()
361 auto diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local
458 T* diff_src_data = in Compute() local
|
| D | mkl_pooling_ops_common.cc | 189 T* diff_src_data, const void* ws_data, in Execute()
|
| D | mkl_avgpooling_op.cc | 279 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
|
| D | mkl_maxpooling_op.cc | 335 T* diff_src_data = output_tensor->flat<T>().data(); in Compute() local
|
| D | mkl_fused_batch_norm_op.cc | 428 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute()
1274 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); in Compute() local
1335 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); in HandleEmptyInput() local
|
| D | mkl_relu_op.cc | 256 void Execute(const T* src_data, const T* diff_dst_data, T* diff_src_data, in Execute()
738 T* diff_src_data = diff_src_tensor->flat<T>().data(); in Compute() local
|