| /external/tensorflow/tensorflow/compiler/xla/ |
| D | sparse_index_array.cc | 24 SparseIndexArray::SparseIndexArray() : rank_(0), max_indices_(0) {} in SparseIndexArray()
28 : indices_(std::move(indices)), rank_(rank), max_indices_(max_indices) { in SparseIndexArray()
29 CHECK_GT(rank_, 0); in SparseIndexArray()
30 CHECK_EQ(indices_.size() % rank_, 0) in SparseIndexArray()
31 << "indices_.size(): " << indices_.size() << ", rank_: " << rank_; in SparseIndexArray()
46 CHECK_GT(rank_, 0); in index_count()
47 CHECK_EQ(indices_.size() % rank_, 0); in index_count()
48 return indices_.size() / rank_; in index_count()
53 CHECK_GT(rank_, 0); in At()
55 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size()); in At()
[all …]
|
| D | sparse_index_array.h | 95 int64 rank() const { return rank_; } in rank()
122 int64 rank_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | union_find.h | 28 UnionFind() : rank_(0), size_(1), parent_(nullptr) {} in UnionFind()
45 int rank_; variable
56 if (a->rank_ > b->rank_) { in Merge()
63 if (a->rank_ == b->rank_) { in Merge()
64 b->rank_++; in Merge()
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 41 rank_(input_shape.dims()), in InputIndexer()
56 index_factors_.resize(rank_); in InputIndexer()
58 for (auto r = rank_ - 1; r >= 0; --r) { in InputIndexer()
94 const int rank_; variable
106 for (auto r = rank_ - 1; r >= 0; --r) { in MoveToOutputIndex()
113 for (int qi = 0; qi < rank_; ++qi) { in MoveToOutputIndex()
121 for (auto r = rank_ - 1; r >= 0; --r) { in MoveToOutputIndex()
128 for (auto r = rank_ - 1; r >= 0; --r) { in MoveToOutputIndex()
135 for (auto r = rank_ - 1; r >= 0; --r) { in IncrementOutputIndex()
192 std::vector<tensorflow::int64> cumulative_dimensions(rank_); in ComputeCumulativeDimensions()
[all …]
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| D | segmentation_ops.h | 72 rank_(rank) {} in BlockedImageUnionFindFunctor()
162 OutputType* const rank_; variable
208 const OutputType rank_a = rank_[index_a]; in do_union()
209 const OutputType rank_b = rank_[index_b]; in do_union()
219 rank_[parent]++; in do_union()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 89 const int32 rank_; variable
343 if (s->rank_ == kUnknownRank) { in Dim()
350 CHECK_NE(s->rank_, kUnknownRank); in DimKnownRank()
359 return s.IsSet() ? s->rank_ : kUnknownRank; in Rank()
802 inline Shape::Shape() : rank_(InferenceContext::kUnknownRank) {} in Shape()
804 : rank_(dims.size()), dims_(dims) {} in Shape()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf_test.cc | 139 rank_(rank) { in BaseSVDFOpModel()
195 int rank_; member in tflite::__anon2a1e039b0111::BaseSVDFOpModel
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 255 if (x_root_member.rank_ < y_root_member.rank_) { in Merge()
264 } else if (x_root_member.rank_ > y_root_member.rank_) { in Merge()
280 ++x_root_member.rank_; in Merge()
|
| D | colocation_graph.h | 91 int rank_ = 0; variable
|
| D | ring_gatherer_test.cc | 383 rank_(rank) { in DeviceInstance()
504 int rank_; member in tensorflow::RingGathererTest::DeviceInstance
|
| D | ring_reducer_test.cc | 407 rank_(rank) { in DeviceInstance()
532 int rank_; member in tensorflow::RingReducerTest::DeviceInstance
|
| D | hierarchical_tree_broadcaster_test.cc | 559 rank_(rank) { in DeviceInstance()
701 int rank_; member in tensorflow::__anonb5878a120111::HierarchicalTreeBroadcasterTest::DeviceInstance
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 218 : parent_(parent), device_name_(device_name), rank_(rank) { in DeviceInstance()
307 int rank_; member in tensorflow::NcclReducerTest::DeviceInstance
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | pattern_matcher.h | 832 explicit constexpr ShapePatternRankImpl(int64 rank) : rank_(rank) {}
835 if (shape->rank() != rank_) {
836 if (rank_ == 0) {
839 EXPLAIN << "Shape does not have rank " << rank_;
847 if (rank_ == 0) {
850 *os << "that has " << rank_ << " dimension" << (rank_ != 1 ? "s" : "");
855 int64 rank_;
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2093 rank_(rank) { in BaseSVDFOpModel()
2153 int rank_; member in tflite::__anon0c68fcb70111::BaseSVDFOpModel
|