| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | inplace_ops.cc | 47 const TensorShape v_shape = ctx->InputShape(2); in Compile() local
53 OP_REQUIRES(ctx, (x_shape.dims() == v_shape.dims()), in Compile()
57 " V.shape=", v_shape.DebugString())); in Compile()
107 const TensorShape v_shape = ctx->InputShape(2); in Compile() local
112 OP_REQUIRES(ctx, (x_shape.dims() == v_shape.dims()), in Compile()
116 " V.shape=", v_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_toeplitz.py | 186 v_shape = array_ops.broadcast_static_shape(
188 return v_shape.concatenate(v_shape[-1:])
193 v_shape = array_ops.broadcast_dynamic_shape(
196 k = v_shape[-1]
197 return array_ops.concat((v_shape, [k]), 0)
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | svd_op_gpu.cu.cc | 126 TensorShape u_shape, v_shape; in RunSVD() local
135 v_shape = shapeRaw; in RunSVD()
136 v_shape.AddDim(n); in RunSVD()
137 v_shape.AddDim(n); in RunSVD()
140 v_shape = V->shape(); in RunSVD()
147 v_shape = shapeRaw; in RunSVD()
148 v_shape.AddDim(p); in RunSVD()
149 v_shape.AddDim(n); in RunSVD()
157 solver->allocate_scoped_tensor(V->dtype(), v_shape, &v_copy), done); in RunSVD()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | linalg_ops.cc | 174 ShapeHandle v_shape; in SelfAdjointEigV2ShapeFn() local
175 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(n, n), &v_shape)); in SelfAdjointEigV2ShapeFn()
176 c->set_output(1, v_shape); in SelfAdjointEigV2ShapeFn()
259 ShapeHandle v_shape; in SvdShapeFn() local
266 c->Concatenate(batch_shape, c->Matrix(n, n), &v_shape)); in SvdShapeFn()
271 c->Concatenate(batch_shape, c->Matrix(n, p), &v_shape)); in SvdShapeFn()
274 c->set_output(2, v_shape); in SvdShapeFn()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util.py | 262 v_shape = v.get_shape().as_list()
271 new_col_vocab_size = v_shape[1]
310 init(shape=v_shape, partition_info=partition_info))
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 76 Shape v_shape = builder->GetShape(result.v).ValueOrDie(); in ComputeMatmulUDVT() local
79 int64 n = ShapeUtil::GetDimension(v_shape, -1); in ComputeMatmulUDVT()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 476 TensorShape var_shape, m_shape, v_shape; in Compile() local
480 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(2, dtype_, &v_shape, &v)); in Compile()
513 OP_REQUIRES(ctx, var_shape.IsSameSize(v_shape), in Compile()
516 v_shape.DebugString())); in Compile()
572 TensorShape var_shape, m_shape, v_shape; in Compile() local
576 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(2, dtype_, &v_shape, &v)); in Compile()
604 OP_REQUIRES(ctx, var_shape.IsSameSize(v_shape), in Compile()
607 v_shape.DebugString())); in Compile()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 470 v_shape = tensor_shape.TensorShape(input_shape[1])
471 dim = v_shape[-1]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 304 v_shape = array_ops.shape(v)
307 np_utils.less_equal(k, -1 * np_utils.getitem(v_shape, 0)),
308 np_utils.greater_equal(k, np_utils.getitem(v_shape, 1)),
|