| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_aggregate_ops.cc | 335 Tensor* dst_tensor = nullptr; variable
338 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor,
342 dst_tensor->scalar<T>()() = std::plus<float>{}(user_i1, user_i2);
349 Tensor* dst_tensor = nullptr; variable
352 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor,
456 Tensor* dst_tensor = nullptr; variable
479 AllocateOutputSetMklShape(ctx, output_idx, &dst_tensor, output_tf_shape,
481 dst.SetUsrMemDataHandle(dst_tensor);
|
| mkl_concat_op.cc | 749 Tensor* dst_tensor = nullptr; local
762 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst,
764 CHECK_NOTNULL(dst_tensor);
768 dst.SetUsrMem(dst_md, dst_tensor);
[all...] |
| mkl_relu_op.cc | 391 Tensor* dst_tensor = nullptr; variable
438 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, tf_shape_dst,
443 dst.SetUsrMem(dst_md, dst_tensor);
646 Tensor* dst_tensor = nullptr; local
651 AllocateOutputSetMklShape(context, dst_index, &dst_tensor,
653 void* out_o = static_cast<void*>(dst_tensor->flat<T>().data());
709 Tensor* dst_tensor = nullptr; local
714 AllocateOutputSetMklShape(context, dst_index, &dst_tensor,
716 void* out_o = static_cast<void*>(dst_tensor->flat<T>().data());
781 Tensor* dst_tensor = nullptr local
[all...] |
| mkl_fused_batch_norm_op.cc | 760 Tensor* dst_tensor = nullptr; variable
[all...] |
