mkl_fused_batch_norm_op.cc | 134 mkl_shape_output.SetTfLayout(mkl_context.mkl_params.in_dim, 137 mkl_shape_output.SetTfDimOrder(mkl_context.mkl_params.in_dim, 219 size_t in_dim; member in struct:tensorflow::MklFusedBatchNormOp::__anon39842::MklBatchNormParams 244 mkl_params.in_dim = input_in_mkl_format 272 dnnLayoutCreate_F32(&mkl_lt_input, mkl_params.in_dim, [all...] |