HomeSort by relevance Sort by last modified time
    Searched refs:v_shape (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/ops/
linalg_ops.cc 103 ShapeHandle v_shape; local
104 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(n, n), &v_shape));
105 c->set_output(1, v_shape);
188 ShapeHandle v_shape; local
195 c->Concatenate(batch_shape, c->Matrix(n, n), &v_shape));
200 c->Concatenate(batch_shape, c->Matrix(n, p), &v_shape));
203 c->set_output(2, v_shape);
  /external/tensorflow/tensorflow/python/training/
warm_starting_util.py 253 v_shape = v.get_shape().as_list()
263 new_col_vocab_size = v_shape[1]
273 # in the axis = 0 case, we can simply use v_shape[1] because we don't
302 init(shape=v_shape, partition_info=partition_info))
  /external/tensorflow/tensorflow/contrib/solvers/python/ops/
lanczos.py 122 v_shape = v.get_shape()
125 v.set_shape(v_shape)
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 412 TensorShape var_shape, m_shape, v_shape; variable
416 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(2, dtype_, &v_shape, &v));
449 OP_REQUIRES(ctx, var_shape.IsSameSize(v_shape),
452 v_shape.DebugString()));
497 TensorShape var_shape, m_shape, v_shape; variable
501 OP_REQUIRES_OK(ctx, ctx->ReadVariableInput(2, dtype_, &v_shape, &v));
529 OP_REQUIRES(ctx, var_shape.IsSameSize(v_shape),
532 v_shape.DebugString()));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
svd_test.cc 71 Shape v_shape = builder->GetShape(result.v).ValueOrDie(); local
74 int64 n = ShapeUtil::GetDimension(v_shape, -1);

Completed in 178 milliseconds