Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_relu_op.cc598 Tensor* diff_src_tensor = nullptr; in Compute() local
616 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute()
736 tf_shape_diff_src, &diff_src_tensor)); in Compute()
739 T* diff_src_data = diff_src_tensor->flat<T>().data(); in Compute()
810 Tensor* diff_src_tensor = nullptr; in Compute_Scalar() local
817 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
819 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data()); in Compute_Scalar()
881 Tensor* diff_src_tensor = nullptr; in Compute_Scalar() local
888 AllocateOutputSetMklShape(context, diff_src_index, &diff_src_tensor, in Compute_Scalar()
890 void* out_o = static_cast<void*>(diff_src_tensor->flat<T>().data()); in Compute_Scalar()
[all …]
H A Dmkl_conv_grad_input_ops.cc345 Tensor* diff_src_tensor = nullptr; in Compute() local
354 AllocateOutputSetMklShape(context, kOutputIdx, &diff_src_tensor, in Compute()
357 DCHECK(diff_src_tensor != nullptr); in Compute()
360 auto diff_src_data = diff_src_tensor->flat<T>().data(); in Compute()
455 AllocateOutputSetMklShape(context, 0, &diff_src_tensor, diff_src_tf_shape, in Compute()
458 static_cast<T*>(const_cast<T*>(diff_src_tensor->flat<T>().data())); in Compute()
H A Dmkl_fused_batch_norm_op.cc1216 Tensor* diff_src_tensor = nullptr; in Compute() local
1220 &diff_src_tensor); in Compute()
1348 AllocateOutputSetMklShape(context, kDiffSrcIndex, &diff_src_tensor, in Compute()
1357 T* diff_src_data = static_cast<T*>(diff_src_tensor->flat<T>().data()); in Compute()
1411 Tensor** diff_src_tensor) { in HandleEmptyInput() argument
1416 AllocateOutputSetMklShape(context, kDiffSrcIndex, diff_src_tensor, in HandleEmptyInput()
1418 auto diff_src_data = (*diff_src_tensor)->flat<T>().data(); in HandleEmptyInput()
1419 std::fill_n(diff_src_data, (*diff_src_tensor)->shape().num_elements(), in HandleEmptyInput()