Searched refs:q_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 74 Shape q_shape = a_shape; in QRDecomposition() local
75 q_shape.mutable_dimensions().back() = m; in QRDecomposition()
77 Shape qr_shape = ShapeUtil::MakeTupleShape({q_shape, a_shape}); in QRDecomposition()
|
| D | math.cc | 1970 TF_ASSIGN_OR_RETURN(auto q_shape, builder.GetShape(q)); in Zeta()
1971 if (x_shape != q_shape) { in Zeta()
1974 x_shape.ToString(), q_shape.ToString()); in Zeta()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | qr_op_impl.h | 158 TensorShape q_shape = input.shape(); in ComputeAsync() local
159 q_shape.set_dim(ndims - 1, full_matrices_ ? m : min_size); in ComputeAsync()
160 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, q_shape, &q), in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 221 ShapeHandle q_shape; in QrShapeFn() local
226 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(m, m), &q_shape)); in QrShapeFn()
229 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(m, p), &q_shape)); in QrShapeFn()
232 c->set_output(0, q_shape); in QrShapeFn()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5624 auto q_shape = llvm::to_vector<4>(type.getShape()); in matchAndRewrite() local
5625 q_shape.back() = m; in matchAndRewrite()
5627 op.getLoc(), RankedTensorType::get(q_shape, type.getElementType()), in matchAndRewrite()
|