Searched refs:u_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | svd_op_gpu.cu.cc | 126 TensorShape u_shape, v_shape; in RunSVD() local
132 u_shape = shapeRaw; in RunSVD()
133 u_shape.AddDim(m); in RunSVD()
134 u_shape.AddDim(m); in RunSVD()
139 u_shape = U->shape(); in RunSVD()
144 u_shape = shapeRaw; in RunSVD()
145 u_shape.AddDim(p); in RunSVD()
146 u_shape.AddDim(m); in RunSVD()
152 context, solver->allocate_scoped_tensor(U->dtype(), u_shape, &u_copy), in RunSVD()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 75 Shape u_shape = builder->GetShape(result.u).ValueOrDie(); in ComputeMatmulUDVT() local
78 int64 m = ShapeUtil::GetDimension(u_shape, -1); in ComputeMatmulUDVT()
91 int num_dims = u_shape.rank(); in ComputeMatmulUDVT()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 258 ShapeHandle u_shape; in SvdShapeFn() local
264 c->Concatenate(batch_shape, c->Matrix(m, m), &u_shape)); in SvdShapeFn()
269 c->Concatenate(batch_shape, c->Matrix(m, p), &u_shape)); in SvdShapeFn()
273 c->set_output(1, u_shape); in SvdShapeFn()
|