Searched refs:rhs_output (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cwise_ops.cc | 96 auto rhs_output = BroadcastTo(rhs, broadcast_helper.output_shape()); in Broadcast() local
97 if (!rhs_output.ok()) { in Broadcast()
98 xla::XlaOp error = rhs.builder()->ReportError(rhs_output.status()); in Broadcast()
101 return {lhs_output.ValueOrDie(), rhs_output.ValueOrDie()}; in Broadcast()
|
| D | matrix_triangular_solve_op.cc | 101 auto rhs_output = BroadcastTo(rhs, rhs_broadcast_shape.dim_sizes()); in Broadcast() local
102 if (!rhs_output.ok()) { in Broadcast()
103 xla::XlaOp error = rhs.builder()->ReportError(rhs_output.status()); in Broadcast()
106 return {lhs_output.ValueOrDie(), rhs_output.ValueOrDie()}; in Broadcast()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf-ops.mlir | 4196 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<2x3x5xi32>, tensor<…
4205 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<2x3x5xi32>, tensor<…
4214 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<5x2xi32>, tensor<2x…
4222 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<2xi32>, tensor<i32>…
4230 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<5x2xi32>, tensor<2x…
4238 …%lhs_output, %rhs_output = "tf.XlaBroadcastHelper"(%arg0, %arg1, %0) : (tensor<2x3x5xi32>, tensor<…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 370 … tf.Output, rhs tf.Output, broadcast_dims tf.Output) (lhs_output tf.Output, rhs_output tf.Output) {
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17626 …TF_Quint8, TF_Uint16, TF_Uint32, TF_Uint64, TF_Uint8]>, [{the broadcasted RHS tensor}]>:$rhs_output
|