OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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