Home
last modified time | relevance | path

Searched refs:CommonFactors (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dutil_test.cc71 TEST(UtilTest, CommonFactors) { in TEST() argument
98 EXPECT_EQ(test_case.expected, CommonFactors(test_case.a, test_case.b)); in TEST()
Dutil.cc227 absl::InlinedVector<std::pair<int64, int64>, 8> CommonFactors( in CommonFactors() function
279 auto common_factors = CommonFactors(from_sizes, to_sizes); in ConvertDimensionNumbers()
Dutil.h476 absl::InlinedVector<std::pair<int64, int64>, 8> CommonFactors(
Dshape_util.cc1184 auto common_factors = CommonFactors(AsInt64Slice(input_shape.dimensions()), in DimensionsUnmodifiedByReshape()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc135 xla::CommonFactors(input_shape.dim_sizes(), shape.dim_sizes()); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dir_array.cc223 CommonFactors(AsInt64Slice(input_shape.dimensions()), in SourceIndexOfReshape()
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.cc830 auto common_factors = CommonFactors(operand->shape().dimensions(), in HandleReshape()
Ddynamic_padder.cc1405 auto common_factors = CommonFactors(operand->shape().dimensions(), in RewriteDynamicReshape()
Dshape_inference.cc2974 auto common_factors = CommonFactors(operand.dimensions(), new_sizes); in InferReshapeShape()
Dalgebraic_simplifier.cc3929 CommonFactors(reshape->operand(0)->shape().dimensions(), in HandleReshape()