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

  /external/tensorflow/tensorflow/core/kernels/
random_shuffle_op.cc 50 static void IndexedShuffle(const int64 size, const InT& input_mat,
58 output_mat.template chip<0>(i) = input_mat.template chip<0>(permutation[i]);
93 const auto input_mat = input.flat_outer_dims<T>(); variable
96 IndexedShuffle<int32>(size, input_mat, output_mat, uniform);
98 IndexedShuffle<int64>(size, input_mat, output_mat, uniform);
  /external/tensorflow/tensorflow/contrib/factorization/python/ops/
factorization_ops_test_utils.py 98 def calculate_loss(input_mat, row_factors, col_factors, regularization=None,
104 input_mat[i, j] is non zero) is (w0 + row_weights[i]col_weights[j]).
107 input_mat: The input matrix, a SparseTensor of rank 2.
125 row_indices, col_indices = array_ops.split(input_mat.indices,
133 indices=input_mat.indices,
135 dense_shape=input_mat.dense_shape)
143 resid = sparse_ops.sparse_add(input_mat, sp_approx * (-1))

Completed in 71 milliseconds