Home
last modified time | relevance | path

Searched refs:BCastArgsOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbcast_ops.cc26 class BCastArgsOp : public OpKernel { class
28 explicit BCastArgsOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in BCastArgsOp() function in tensorflow::BCastArgsOp
66 TF_DISALLOW_COPY_AND_ASSIGN(BCastArgsOp);
125 BCastArgsOp<int32>);
132 BCastArgsOp<int64>);
139 BCastArgsOp<int32>);
146 BCastArgsOp<int64>);
155 BCastArgsOp<int32>);
162 BCastArgsOp<int32>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbcast_ops.cc32 class BCastArgsOp : public XlaOpKernel { class
34 explicit BCastArgsOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BCastArgsOp() function in tensorflow::__anon019ed33d0111::BCastArgsOp
67 TF_DISALLOW_COPY_AND_ASSIGN(BCastArgsOp);
72 BCastArgsOp);