Searched defs:output_acc (Results 1 – 6 of 6) sorted by relevance
237 mlir::memref::AllocOp output_acc; member266 auto output_acc = builder.create<mlir::memref::AllocOp>( in CreateNaiveMlirConv() local400 mlir::memref::AllocOp output_acc = anchors.output_acc; in TransformMlirConv() local
75 auto output_acc = output.accessor<scalar_t, 1>(); in multi_margin_loss_cpu_kernel() local85 auto output_acc = output.data_ptr<scalar_t>(); in multi_margin_loss_cpu_kernel() local
95 auto output_acc = output.accessor<scalar_t, 1>(); in multilabel_margin_loss_forward_out_frame() local
181 auto output_acc = output.accessor<scalar_t, 1>(); in nll_loss_out_frame() local
125 auto output_acc = output.accessor<scalar_t, 3>(); in nll_loss2d_forward_out_frame() local
261 accscalar_t output_acc = 0; in nll_loss_forward_reduce_cuda_kernel_2d() local