Searched refs:common_factors (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 135 std::vector<std::pair<int64, int64>> common_factors = in SourceIndexOfReshape() local
142 for (ssize_t k = common_factors.size() - 2; k >= 0; --k) { in SourceIndexOfReshape()
145 common_factors[k].second, in SourceIndexOfReshape()
146 common_factors[k + 1].second - common_factors[k].second), in SourceIndexOfReshape()
149 .subspan(common_factors[k].second, in SourceIndexOfReshape()
150 common_factors[k + 1].second - in SourceIndexOfReshape()
151 common_factors[k].second), in SourceIndexOfReshape()
156 for (int64 i = common_factors[k + 1].first - 1; in SourceIndexOfReshape()
157 i >= common_factors[k].first; --i) { in SourceIndexOfReshape()
162 } else if (i == common_factors[k].first) { in SourceIndexOfReshape()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1055 auto common_factors = CommonFactors(AsInt64Slice(input_shape.dimensions()), in DimensionsUnmodifiedByReshape() local
1057 for (size_t i = 0; i < common_factors.size() - 1;) { in DimensionsUnmodifiedByReshape()
1058 if (1 != common_factors[i + 1].first - common_factors[i].first || in DimensionsUnmodifiedByReshape()
1059 1 != common_factors[i + 1].second - common_factors[i].second) { in DimensionsUnmodifiedByReshape()
1060 common_factors.erase(common_factors.begin() + i); in DimensionsUnmodifiedByReshape()
1066 common_factors.pop_back(); in DimensionsUnmodifiedByReshape()
1067 return common_factors; in DimensionsUnmodifiedByReshape()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 691 std::set<string> common_factors; in TrySimplify() local
694 node, &common_factors, &common_factor_is_denominator, &ctrl_deps)); in TrySimplify()
696 if (common_factors.size() == 1) { in TrySimplify()
697 const string& common_factor = *common_factors.begin(); in TrySimplify()
767 Status GetCommonFactors(const NodeDef* node, std::set<string>* common_factors, in GetCommonFactors() argument
770 CHECK(common_factors->empty()); in GetCommonFactors()
777 if (i > 0 && common_factors->empty()) break; in GetCommonFactors()
789 common_factors->clear(); in GetCommonFactors()
803 common_factors->clear(); in GetCommonFactors()
816 std::swap(*common_factors, factors_i); in GetCommonFactors()
[all …]
|