OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:mask_indices
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/python/kernel_tests/
sparsemask_op_test.py
32
mask_indices
= np.array([0], dtype=np.int32)
40
mask_indices_tensor = ops.convert_to_tensor(
mask_indices
)
/external/tensorflow/tensorflow/contrib/factorization/kernels/
masked_matmul_ops.cc
68
// Computes the product a * b, but only for indices (i, j) in
mask_indices
.
70
// prod_values[i] = (a * b)[
mask_indices
[i, 0],
mask_indices
[i, 1]].
73
const Tensor&
mask_indices
= context->input(2);
variable
81
OP_REQUIRES(context, TensorShapeUtils::IsMatrix(
mask_indices
.shape()),
82
InvalidArgument("Input
mask_indices
should be a matrix."));
94
const int64 num_nonzero_elements =
mask_indices
.dim_size(0);
100
OP_REQUIRES(context,
mask_indices
.dim_size(1) == 2,
101
InvalidArgument("
mask_indices
should be a matrix of shape ",
109
ConstEigenMatInt64Map indices_mat(
mask_indices
.matrix<int64>().data()
[
all
...]
/external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/
masked_matmul_benchmark.py
99
a, b,
mask_indices
= session.run([a_var, b_var, mask.indices])
101
mask_indices_ph:
mask_indices
,
/external/tensorflow/tensorflow/python/ops/
array_ops.py
[
all
...]
Completed in 222 milliseconds