Searched refs:AllocateTFOutputs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 791 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
803 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
958 AllocateTFOutputs(context, tf_shape_scale, workspace_tf_shape, in HandleEmptyInput()
964 void AllocateTFOutputs(OpKernelContext* context, TensorShape tf_shape_scale, in AllocateTFOutputs() function in tensorflow::MklFusedBatchNormOp
1282 AllocateTFOutputs(context, scale_tensor.shape(), &diff_scale_tensor, in Compute()
1331 AllocateTFOutputs(context, tf_shape_scale_shift, &diff_scale_tensor, in HandleEmptyInput()
1335 void AllocateTFOutputs(OpKernelContext* context, in AllocateTFOutputs() function in tensorflow::MklFusedBatchNormGradOp
|