Searched refs:q_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr_test.cc | 66 TF_ASSERT_OK_AND_ASSIGN(xla::Shape q_shape, builder.GetShape(q)); in XLA_TEST_F()
68 EXPECT_EQ(q_shape, in XLA_TEST_F()
|
| D | math.cc | 1978 TF_ASSIGN_OR_RETURN(auto q_shape, builder.GetShape(q)); in Zeta()
1979 if (x_shape != q_shape) { in Zeta()
1982 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/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 500 q_shape = array_ops.shape(q)
504 q = math_ops.cumsum(q_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 6541 auto q_shape = llvm::to_vector<4>(type.getShape()); in matchAndRewrite() local
6542 q_shape.back() = m; in matchAndRewrite()
|