Searched refs:ix_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | dim_comparator.h | 51 : ix_(ix), order_(order), dims_(shape.size()) { in DimComparator()
63 if (ix_(i, d) < ix_(j, d)) return true; in operator()
64 if (ix_(i, d) > ix_(j, d)) return false; in operator()
88 const TTypes<int64>::Matrix ix_;
106 if (ix_(i, d) < ix_(j, d)) { in operator()
110 if (ix_(i, d) > ix_(j, d)) break; in operator()
|
| D | group_iterator.h | 82 : ix_(ix), in GroupIterable()
83 ix_matrix_(ix_.matrix<int64>()), in GroupIterable()
92 CHECK(loc >= 0 && loc <= ix_.dim_size(0)) in at()
93 << "loc provided must lie between 0 and " << ix_.dim_size(0); in at()
96 IteratorStep end() { return IteratorStep(this, ix_.dim_size(0)); } in end()
131 const Tensor ix_; variable
|
| D | sparse_tensor.cc | 68 result->ix_ = std::move(ix); in Create()
100 : ix_(std::move(ix)), in SparseTensor()
104 dims_(UnsafeGetDimsFromIx(ix_)) { in SparseTensor()
105 DCHECK_EQ(ix_.dtype(), DT_INT64) in SparseTensor()
106 << "indices must be type int64 but got: " << ix_.dtype(); in SparseTensor()
109 DCHECK_EQ(ix_.shape().dim_size(0), vals_.shape().dim_size(0)) in SparseTensor()
133 const auto ix_t = ix_.matrix<int64>(); in IndicesValidVectorFastPath()
155 const auto ix_t = ix_.matrix<int64>(); in IndicesValidMatrix32BitFastPath()
226 const auto ix_t = ix_.matrix<int64>(); in IndicesValidHelper()
|
| D | sparse_tensor.h | 81 : SparseTensor(other.ix_, other.vals_, other.shape_, other.order_) {} in SparseTensor()
84 : SparseTensor(std::move(other.ix_), std::move(other.vals_), in SparseTensor()
88 ix_ = other.ix_;
97 ix_ = std::move(other.ix_);
105 std::size_t num_entries() const { return ix_.dim_size(0); } in num_entries()
109 const Tensor& indices() const { return ix_; } in indices()
139 return GroupIterable(ix_, vals_, dims_, group_ix); in group()
261 Tensor ix_; variable
276 auto ix_t = ix_.matrix<int64>(); in Reorder()
365 auto ix_t = ix_.matrix<int64>(); in ToDense()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops_test.py | 1152 tnp_ans = np_array_ops.ix_(*arys)
1153 onp_ans = np.ix_(*arys)
|
| D | np_array_ops.py | 1337 def ix_(*args): # pylint: disable=missing-docstring function
|
| /external/tensorflow/third_party/py/numpy/tf_numpy_api/ |
| D | tensorflow.experimental.numpy.pbtxt | 536 name: "ix_"
|