Searched refs:u_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | svd_op_gpu.cu.cc | 106 TensorShape u_shape; in RunSVD() local
108 u_shape = U->shape(); in RunSVD()
112 u_shape = shapeRaw; in RunSVD()
113 u_shape.AddDim(p); in RunSVD()
114 u_shape.AddDim(m); in RunSVD()
117 context, solver->allocate_scoped_tensor(U->dtype(), u_shape, &u_copy), in RunSVD()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 70 Shape u_shape = builder->GetShape(result.u).ValueOrDie(); in ComputeMatmulUDVT() local
73 int64 m = ShapeUtil::GetDimension(u_shape, -1); in ComputeMatmulUDVT()
87 int num_dims = u_shape.rank(); in ComputeMatmulUDVT()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 187 ShapeHandle u_shape; in SvdShapeFn() local
193 c->Concatenate(batch_shape, c->Matrix(m, m), &u_shape)); in SvdShapeFn()
198 c->Concatenate(batch_shape, c->Matrix(m, p), &u_shape)); in SvdShapeFn()
202 c->set_output(1, u_shape); in SvdShapeFn()
|