Searched refs:AllocateOutput (Results 1 – 1 of 1) sorted by relevance
206 ctx, AllocateOutput(ctx, rank, batch_size, num_lhs_rows, num_rhs_cols, in Compute()236 Status AllocateOutput(OpKernelContext* ctx, const int32 rank, in AllocateOutput() function in tensorflow::CSRMatMulCPUOp