HomeSort by relevance Sort by last modified time
    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