Home
last modified time | relevance | path

Searched refs:common_factors (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc140 const auto common_factors = in SourceIndexOfReshape() local
147 for (ssize_t k = common_factors.size() - 2; k >= 0; --k) { in SourceIndexOfReshape()
150 .subspan(common_factors[k].second, in SourceIndexOfReshape()
151 common_factors[k + 1].second - common_factors[k].second); in SourceIndexOfReshape()
154 common_factors[k].second, in SourceIndexOfReshape()
155 common_factors[k + 1].second - common_factors[k].second), in SourceIndexOfReshape()
161 for (int64 i = common_factors[k + 1].first - 1; in SourceIndexOfReshape()
162 i >= common_factors[k].first; --i) { in SourceIndexOfReshape()
167 } else if (i == common_factors[k].first) { in SourceIndexOfReshape()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1055 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 std::vector<std::pair<int64, int64>>(common_factors.begin(), in DimensionsUnmodifiedByReshape()
1068 common_factors.end()); in DimensionsUnmodifiedByReshape()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc579 auto common_factors = CommonFactors(operand->shape().dimensions(), in HandleReshape() local
586 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in HandleReshape()
587 auto start = common_factors[i]; in HandleReshape()
588 auto end = common_factors[i + 1]; in HandleReshape()
628 for (auto common_factor : common_factors) { in HandleReshape()
Ddynamic_padder.cc508 auto common_factors = in RewriteDynamicReshapeSingleDim() local
520 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in RewriteDynamicReshapeSingleDim()
521 auto start = common_factors[i]; in RewriteDynamicReshapeSingleDim()
522 auto end = common_factors[i + 1]; in RewriteDynamicReshapeSingleDim()
Dshape_inference.cc2831 auto common_factors = CommonFactors(operand.dimensions(), new_sizes); in InferReshapeShape() local
2848 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in InferReshapeShape()
2849 auto start = common_factors[i]; in InferReshapeShape()
2850 auto end = common_factors[i + 1]; in InferReshapeShape()
2873 for (auto common_factor : common_factors) { in InferReshapeShape()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc714 std::set<string> common_factors; in TrySimplify() local
717 node, &common_factors, &common_factor_is_denominator, &ctrl_deps)); in TrySimplify()
719 if (common_factors.size() == 1) { in TrySimplify()
720 const string& common_factor = *common_factors.begin(); in TrySimplify()
790 Status GetCommonFactors(const NodeDef* node, std::set<string>* common_factors, in GetCommonFactors() argument
793 CHECK(common_factors->empty()); in GetCommonFactors()
800 if (i > 0 && common_factors->empty()) break; in GetCommonFactors()
812 common_factors->clear(); in GetCommonFactors()
827 common_factors->clear(); in GetCommonFactors()
840 std::swap(*common_factors, factors_i); in GetCommonFactors()
[all …]