Searched defs:AllocateTFOutputs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ | ||
| D | mkl_fused_batch_norm_op.cc | 964 void AllocateTFOutputs(OpKernelContext* context, TensorShape tf_shape_scale, in AllocateTFOutputs() function in tensorflow::MklFusedBatchNormOp |