Searched refs:tf_shape_scale (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 941 TensorShape tf_shape_scale, Tensor** dst_tensor) { in HandleEmptyInput() argument
958 AllocateTFOutputs(context, tf_shape_scale, workspace_tf_shape, in HandleEmptyInput()
964 void AllocateTFOutputs(OpKernelContext* context, TensorShape tf_shape_scale, in AllocateTFOutputs() argument
986 tf_shape_scale, mkl_shape_batch_mean, in AllocateTFOutputs()
991 int num_elements = tf_shape_scale.num_elements(); in AllocateTFOutputs()
999 batch_variance_tensor, tf_shape_scale, in AllocateTFOutputs()
1011 tf_shape_scale, mkl_shape_saved_mean, in AllocateTFOutputs()
1022 saved_variance_tensor, tf_shape_scale, in AllocateTFOutputs()
|