Home
last modified time | relevance | path

Searched refs:diff_src_tensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_relu_op.cc596 Tensor* diff_src_tensor = nullptr; in Compute() local
614 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute()
733 tf_shape_diff_src, &diff_src_tensor)); in Compute()
736 T* diff_src_data = diff_src_tensor->flat<T>().data(); in Compute()
807 Tensor* diff_src_tensor = nullptr; in Compute_Scalar() local
814 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
816 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data()); in Compute_Scalar()
879 Tensor* diff_src_tensor = nullptr; in Compute_Scalar() local
886 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
888 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data()); in Compute_Scalar()
[all …]
Dmkl_conv_grad_input_ops.cc346 Tensor* diff_src_tensor = nullptr; in Compute() local
355 AllocateOutputSetMklShape(context, kOutputIdx, &diff_src_tensor, in Compute()
358 DCHECK(diff_src_tensor != nullptr); in Compute()
361 auto diff_src_data = diff_src_tensor->flat<T>().data(); in Compute()
454 AllocateOutputSetMklShape(context, 0, &diff_src_tensor, diff_src_tf_shape, in Compute()
457 static_cast<T*>(const_cast<T*>(diff_src_tensor->flat<T>().data())); in Compute()
Dmkl_fused_batch_norm_op.cc1125 Tensor* diff_src_tensor = nullptr; in Compute() local
1129 &diff_src_tensor); in Compute()
1256 AllocateOutputSetMklShape(context, kDiffSrcIndex, &diff_src_tensor, in Compute()
1265 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); in Compute()
1318 Tensor** diff_src_tensor) { in HandleEmptyInput() argument
1323 AllocateOutputSetMklShape(context, kDiffSrcIndex, diff_src_tensor, in HandleEmptyInput()
1325 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); in HandleEmptyInput()
1326 std::fill_n(diff_src_data, (*diff_src_tensor)->shape().num_elements(), in HandleEmptyInput()