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

  /external/tensorflow/tensorflow/python/ops/
linalg_ops.py 70 small_dim = matrix_shape[-1]
72 small_dim = matrix_shape[-2]
73 identity = eye(small_dim, batch_shape=batch_shape, dtype=matrix.dtype)
  /external/tensorflow/tensorflow/core/kernels/
batch_matmul_op_impl.h 211 const int64 small_dim = std::min( local
215 if (small_dim > 1 &&

Completed in 173 milliseconds