| /external/tensorflow/tensorflow/compiler/xla/ |
| 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/llvm-project/flang/lib/Semantics/ |
| D | check-allocate.cpp | 44 allocateShapeSpecRank_{ShapeSpecRank(alloc)}, rank_{symbol_
94 const int rank_{0}; member in Fortran::semantics::AllocationCheckerHelper
485 if (rank_ > 0) { in RunChecks()
493 if (allocateInfo_.sourceExprRank != rank_) { in RunChecks()
500 allocateInfo_.sourceExprRank, rank_); in RunChecks()
506 if (allocateShapeSpecRank_ != rank_) { in RunChecks()
510 .Attach(symbol_->name(), "Declared here with rank %d"_en_US, rank_); in RunChecks()
524 allocateInfo_.sourceExprRank != rank_) { in RunChecks()
531 "Allocatable object declared here with rank %d"_en_US, rank_); in RunChecks()
|
| D | symbol.cpp | 125 void AssocEntityDetails::set_rank(int rank) { rank_ = rank; } in set_rank()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | svdf_test.cc | 140 rank_(rank) { in BaseSVDFOpModel()
196 int rank_; member in tflite::__anon7126f9820111::BaseSVDFOpModel
476 rank_(rank) { in IntegerSVDFOpModel()
533 int rank_; member in tflite::__anon7126f9820111::IntegerSVDFOpModel
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference.h | 85 const int32 rank_; variable
330 if (!s.Handle() || s->rank_ == kUnknownRank) { in Dim()
337 CHECK_NE(s->rank_, kUnknownRank); in DimKnownRank()
345 return s.IsSet() ? s->rank_ : kUnknownRank; in Rank()
787 inline Shape::Shape() : rank_(InferenceContext::kUnknownRank) {} in Shape()
789 : rank_(dims.size()), dims_(dims) {} in Shape()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | colocation_graph.cc | 310 if (x_root_member.rank_ < y_root_member.rank_) { in Merge()
319 } else if (x_root_member.rank_ > y_root_member.rank_) { in Merge()
335 ++x_root_member.rank_; in Merge()
|
| D | colocation_graph.h | 128 int rank_ = 0; variable
|
| D | permuter_test.cc | 326 rank_(rank), in DeviceInstance()
419 int rank_; member in tensorflow::__anonfb3b1ff60111::PermuterTest::DeviceInstance
|
| D | ring_gatherer_test.cc | 393 rank_(rank), in DeviceInstance()
512 int rank_; member in tensorflow::RingGathererTest::DeviceInstance
|
| D | ring_reducer_test.cc | 419 rank_(rank), in DeviceInstance()
543 int rank_; member in tensorflow::RingReducerTest::DeviceInstance
|
| D | hierarchical_tree_broadcaster_test.cc | 565 rank_(rank), in DeviceInstance()
702 int rank_; member in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest::DeviceInstance
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | pattern_matcher.h | 813 explicit constexpr ShapePatternRankImpl(int64 rank) : rank_(rank) {}
816 if (shape->rank() != rank_) {
817 if (rank_ == 0) {
820 EXPLAIN << "Shape does not have rank " << rank_;
828 if (rank_ == 0) {
831 *os << "that has " << rank_ << " dimension" << (rank_ != 1 ? "s" : "");
836 int64 rank_;
|
| /external/llvm-project/flang/unittests/Evaluate/ |
| D | ISO-Fortran-binding.cpp | 24 MATCH(sizeof(dvStorage_), Descriptor::SizeInBytes(rank_, false)); in Check()
26 MATCH(true, sizeof(dvStorage_) >= Descriptor::SizeInBytes(rank_, false)); in Check()
35 static constexpr int rank_{rank}; member in Test_CFI_CDESC_T
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 252 rank_(rank), in DeviceInstance()
422 int rank_; member in tensorflow::NcclTestBase::DeviceInstance
|
| /external/llvm-project/flang/include/flang/Semantics/ |
| D | symbol.h | 152 std::optional<int> rank() const { return rank_; } in rank()
156 std::optional<int> rank_; variable
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2927 rank_(rank) { in BaseSVDFOpModel()
2986 int rank_; member in tflite::__anonde5f765e0111::BaseSVDFOpModel
|