Searched refs:mask_indices (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 73 const Tensor& mask_indices = context->input(2); in Compute() local
81 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(mask_indices.shape()), in Compute()
94 const int64 num_nonzero_elements = mask_indices.dim_size(0); in Compute()
100 OP_REQUIRES(context, mask_indices.dim_size(1) == 2, in Compute()
109 ConstEigenMatInt64Map indices_mat(mask_indices.matrix<int64>().data(), in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparsemask_op_test.py | 34 mask_indices = np.array([0], dtype=np.int32)
42 mask_indices_tensor = ops.convert_to_tensor(mask_indices)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | 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/tools/api/golden/v2/ |
| D | tensorflow.sparse.pbtxt | 41 …argspec: "args=[\'a\', \'mask_indices\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.pbtxt | 41 …argspec: "args=[\'a\', \'mask_indices\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
|
| D | tensorflow.pbtxt | 2077 …argspec: "args=[\'a\', \'mask_indices\', \'name\'], varargs=None, keywords=None, defaults=[\'None\…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1405 def sparse_mask(a, mask_indices, name=None): argument
1440 with ops.name_scope(name, "sparse_mask", [a, mask_indices]) as name:
1442 out_indices, to_gather = setdiff1d(indices, mask_indices)
|