Searched refs:BCastGradArgsOp (Results 1 – 2 of 2) sorted by relevance
75 class BCastGradArgsOp : public OpKernel { class77 explicit BCastGradArgsOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in BCastGradArgsOp() function in tensorflow::BCastGradArgsOp116 TF_DISALLOW_COPY_AND_ASSIGN(BCastGradArgsOp);172 BCastGradArgsOp<int32>);180 BCastGradArgsOp<int64>);188 BCastGradArgsOp<int32>);196 BCastGradArgsOp<int64>);206 BCastGradArgsOp<int32>);214 BCastGradArgsOp<int64>);
79 class BCastGradArgsOp : public XlaOpKernel { class81 explicit BCastGradArgsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BCastGradArgsOp() function in tensorflow::__anon019ed33d0111::BCastGradArgsOp121 TF_DISALLOW_COPY_AND_ASSIGN(BCastGradArgsOp);127 BCastGradArgsOp);