Searched refs:tf_shape_src (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 691 TensorShape tf_shape_src; in Compute() local
696 tf_shape_src = dnn_shape_src.GetTfShape(); in Compute()
701 tf_shape_src = src_tensor.shape(); in Compute()
724 if (tf_shape_src.num_elements() == 0) { in Compute()
727 HandleEmptyInput(context, tf_shape_src, workspace_tf_shape, in Compute()
939 void HandleEmptyInput(OpKernelContext* context, TensorShape tf_shape_src, in HandleEmptyInput() argument
947 AllocateOutputSetMklShape(context, kDstIndex, dst_tensor, tf_shape_src, in HandleEmptyInput()
1084 TensorShape tf_shape_src, tf_shape_diff_dst; in Compute() local
1100 tf_shape_src = dnn_shape_src.GetTfShape(); in Compute()
1105 tf_shape_src = src_tensor.shape(); in Compute()
[all …]
|