Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc279 auto common_factors = CommonFactors(from_sizes, to_sizes); in ConvertDimensionNumbers() local
280 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in ConvertDimensionNumbers()
283 for (int64 d = common_factors[i].first; d < common_factors[i + 1].first; in ConvertDimensionNumbers()
290 for (int64 d = common_factors[i].second; d < common_factors[i + 1].second; in ConvertDimensionNumbers()
294 for (int64 d = common_factors[i].first; d < common_factors[i + 1].first; in ConvertDimensionNumbers()
299 for (int64 d = common_factors[i].first; d < common_factors[i + 1].first; in ConvertDimensionNumbers()
Dshape_util.cc1184 auto common_factors = CommonFactors(AsInt64Slice(input_shape.dimensions()), in DimensionsUnmodifiedByReshape() local
1186 for (size_t i = 0; i < common_factors.size() - 1;) { in DimensionsUnmodifiedByReshape()
1187 if (1 != common_factors[i + 1].first - common_factors[i].first || in DimensionsUnmodifiedByReshape()
1188 1 != common_factors[i + 1].second - common_factors[i].second) { in DimensionsUnmodifiedByReshape()
1189 common_factors.erase(common_factors.begin() + i); in DimensionsUnmodifiedByReshape()
1195 common_factors.pop_back(); in DimensionsUnmodifiedByReshape()
1196 return std::vector<std::pair<int64, int64>>(common_factors.begin(), in DimensionsUnmodifiedByReshape()
1197 common_factors.end()); in DimensionsUnmodifiedByReshape()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc134 auto common_factors = in Compile() local
138 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in Compile()
139 auto start = common_factors[i]; in Compile()
140 auto end = common_factors[i + 1]; in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc222 const auto common_factors = in SourceIndexOfReshape() local
227 for (ssize_t k = common_factors.size() - 2; k >= 0; --k) { in SourceIndexOfReshape()
230 .subspan(common_factors[k].second, in SourceIndexOfReshape()
231 common_factors[k + 1].second - common_factors[k].second); in SourceIndexOfReshape()
234 common_factors[k].second, in SourceIndexOfReshape()
235 common_factors[k + 1].second - common_factors[k].second), in SourceIndexOfReshape()
241 for (int64 i = common_factors[k + 1].first - 1; in SourceIndexOfReshape()
242 i >= common_factors[k].first; --i) { in SourceIndexOfReshape()
247 } else if (i == common_factors[k].first) { in SourceIndexOfReshape()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc717 std::set<string> common_factors; in TrySimplify() local
720 node, &common_factors, &common_factor_is_denominator, &ctrl_deps)); in TrySimplify()
722 if (common_factors.size() == 1) { in TrySimplify()
723 const string& common_factor = *common_factors.begin(); in TrySimplify()
793 Status GetCommonFactors(const NodeDef* node, std::set<string>* common_factors, in GetCommonFactors() argument
796 CHECK(common_factors->empty()); in GetCommonFactors()
803 if (i > 0 && common_factors->empty()) break; in GetCommonFactors()
815 common_factors->clear(); in GetCommonFactors()
830 common_factors->clear(); in GetCommonFactors()
843 std::swap(*common_factors, factors_i); in GetCommonFactors()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc830 auto common_factors = CommonFactors(operand->shape().dimensions(), in HandleReshape() local
837 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in HandleReshape()
838 auto start = common_factors[i]; in HandleReshape()
839 auto end = common_factors[i + 1]; in HandleReshape()
879 for (auto common_factor : common_factors) { in HandleReshape()
Ddynamic_padder.cc1405 auto common_factors = CommonFactors(operand->shape().dimensions(), in RewriteDynamicReshape() local
1408 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in RewriteDynamicReshape()
1409 auto start = common_factors[i]; in RewriteDynamicReshape()
1410 auto end = common_factors[i + 1]; in RewriteDynamicReshape()
Dshape_inference.cc2974 auto common_factors = CommonFactors(operand.dimensions(), new_sizes); in InferReshapeShape() local
2991 for (int64 i = 0; i < common_factors.size() - 1; ++i) { in InferReshapeShape()
2992 auto start = common_factors[i]; in InferReshapeShape()
2993 auto end = common_factors[i + 1]; in InferReshapeShape()
3016 for (auto common_factor : common_factors) { in InferReshapeShape()
Dalgebraic_simplifier.cc3928 auto common_factors = in HandleReshape() local
3932 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape()
3937 common_factors, [&](const std::pair<int64, int64>& dim_pair) { in HandleReshape()
3940 if (iota_dim != common_factors.end() && next_dim != common_factors.end()) { in HandleReshape()