Searched refs:neg_beta (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 6044 auto neg_beta = builder->create<NegOp>(loc, beta); in ComputeWYRepresentation() local
6049 loc, neg_beta, v_wyv, in ComputeWYRepresentation()
6065 auto neg_beta = rewriter->create<NegOp>(loc, beta); in ComputeWYRepresentation() local
6069 loc, neg_beta, v, GetI64ElementsAttr(beta_broadcast_dims, rewriter), in ComputeWYRepresentation()
|