Searched refs:tf_shape_dst (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 722 TensorShape tf_shape_dst; in Compute() local
729 tf_shape_dst.AddDim((dst_pd.get_size() / sizeof(T))); in Compute()
730 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
761 TensorShape tf_shape_dst; in Compute() local
764 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
765 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
798 TensorShape tf_shape_dst; in Compute() local
801 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
803 AllocateOutputSetMklShape(context, 0, &dst_tensor, tf_shape_dst, in Compute()
|
| D | mkl_relu_op.cc | 451 TensorShape tf_shape_dst; in Compute() local
456 tf_shape_dst = MklGetInput(context, src_index).shape(); in Compute()
457 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, tf_shape_dst, in Compute()
512 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T)); in Compute()
517 tf_shape_dst = src_tensor.shape(); in Compute()
522 AllocateOutputSetMklShape(context, dst_index, &dst_tensor, tf_shape_dst, in Compute()
531 tf_shape_dst, &dst_tensor)); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 847 TensorShape tf_shape_dst; in Compute() local
855 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T)); in Compute()
857 tf_shape_dst = dnn_shape_dst.GetTfShape(); in Compute()
859 AllocateOutputSetMklShape(context, kDstIndex, &dst_tensor, tf_shape_dst, in Compute()
|