Searched refs:ones_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_householder.py | 261 ones_shape = array_ops.concat([result_shape[:-1], [n - 1]], axis=-1)
263 eigvals = array_ops.ones(shape=ones_shape, dtype=self.dtype)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 464 Node* ones_shape; in StackTensor() local
467 .Finalize(g, &ones_shape)); in StackTensor()
471 node_builder("OnesLike").Input(ones_shape).Finalize(g, &ones)); in StackTensor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_impl.py | 1804 ones_shape = array_ops.stack([cols, 1])
1805 ones = array_ops.ones(ones_shape, x.dtype)
|
| D | array_ops.py | 3164 ones_shape = shape_internal(tensor, optimize=optimize)
3167 ret = ones(ones_shape, dtype=dtype, name=name)
|
| D | math_grad.py | 331 ones_shape = array_ops.concat([
335 ones = array_ops.fill(ones_shape, constant_op.constant(1, dtype=grad.dtype))
|