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

  /external/tensorflow/tensorflow/core/kernels/
qr_op_impl.h 155 TensorShape q_shape = input.shape(); variable
156 q_shape.set_dim(ndims - 1, full_matrices_ ? m : min_size);
157 OP_REQUIRES_OK_ASYNC(context, context->allocate_output(0, q_shape, &q),
  /external/tensorflow/tensorflow/core/ops/
linalg_ops.cc 126 ShapeHandle q_shape; local
131 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(m, m), &q_shape));
134 TF_RETURN_IF_ERROR(c->Concatenate(batch_shape, c->Matrix(m, p), &q_shape));
137 c->set_output(0, q_shape);

Completed in 90 milliseconds