Home
last modified time | relevance | path

Searched refs:lhs_size (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_as_convolution_util.cc27 bool ConvSpatialDimensionIsParallel(const WindowDimension& wd, int64 lhs_size) { in ConvSpatialDimensionIsParallel() argument
31 if (lhs_size == wd.size() && lhs_size == wd.base_dilation() && in ConvSpatialDimensionIsParallel()
32 ((std::max<int64>(1, lhs_size - 1) == wd.stride() && in ConvSpatialDimensionIsParallel()
34 (std::max<int64>(1, lhs_size - 1) == wd.window_dilation() && in ConvSpatialDimensionIsParallel()
42 if (wd.size() == lhs_size && wd.padding_high() == lhs_size - 1 && in ConvSpatialDimensionIsParallel()
43 wd.padding_low() == lhs_size - 1 && wd.window_reversal() && in ConvSpatialDimensionIsParallel()
44 wd.window_dilation() == 1 && wd.stride() == lhs_size && in ConvSpatialDimensionIsParallel()
45 wd.base_dilation() == lhs_size - 1) { in ConvSpatialDimensionIsParallel()
69 int64 lhs_size = conv->operand(0)->shape().dimensions(lhs); in ParseConvolutionDimsInfo() local
74 if (ConvSpatialDimensionIsParallel(wd, lhs_size)) { in ParseConvolutionDimsInfo()
[all …]
Ddot_as_convolution_util.h69 bool ConvSpatialDimensionIsParallel(const WindowDimension& wd, int64 lhs_size);
Dalgebraic_simplifier.cc626 const int64 lhs_size = ShapeUtil::ElementsIn(lhs->shape()); in ScalarMultiplyReduction() local
636 if (dot_size <= std::min(lhs_size, rhs_size)) { in ScalarMultiplyReduction()
638 if (dot_size < lhs_size) { in ScalarMultiplyReduction()
644 } else if (lhs_size <= rhs_size) { in ScalarMultiplyReduction()
646 if (lhs_size < rhs_size) { in ScalarMultiplyReduction()
649 if (lhs_size < dot_size && dot->user_count() == 1) { in ScalarMultiplyReduction()
654 if (rhs_size < lhs_size) { in ScalarMultiplyReduction()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_triangular_solve_op.cc53 auto lhs_size = lhs_shape.dims(); in Compile() local
56 lhs_shape.dim_size(lhs_size - 1) == lhs_shape.dim_size(lhs_size - 2), in Compile()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConcatenation.h269 const double lhs_size = m_leftImpl.dimensions().TotalSize();
271 return (lhs_size / (lhs_size + rhs_size)) *
273 (rhs_size / (lhs_size + rhs_size)) *
DTensorContractionThreadPool.h357 size_t lhs_size =
362 (nm0_ * lhs_size + nn0_ * rhs_size) * std::min<size_t>(nk_, P - 1)));
368 mem += lhs_size;
/external/libtextclassifier/native/utils/grammar/utils/
Dir.cc112 const int lhs_size = rules_set->lhs.size(); in AddLhsSet() local
114 for (int i = 0; i < lhs_size; i++) { in AddLhsSet()
131 serialized_lhs_set->lhs.push_back(-lhs_size); in AddLhsSet()
/external/openscreen/discovery/mdns/
Dmdns_records.cc52 const size_t lhs_size = lhs_cast.MaxWireSize() + 2; in IsGreaterThan() local
55 uint8_t lhs_bytes[lhs_size]; in IsGreaterThan()
57 MdnsWriter lhs_writer(lhs_bytes, lhs_size); in IsGreaterThan()
73 return lhs_size > rhs_size; in IsGreaterThan()
/external/openscreen/third_party/abseil/src/absl/strings/
Dcord.cc1407 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1409 if (lhs_size == rhs_size) { in SharedCompareImpl()
1410 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1412 if (lhs_size < rhs_size) { in SharedCompareImpl()
1413 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1261 size_t lhs_size = lhs.size();
1263 if (lhs_size != rhs_size) return false;
/external/webrtc/third_party/abseil-cpp/absl/strings/
Dcord.cc1433 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1435 if (lhs_size == rhs_size) { in SharedCompareImpl()
1436 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1438 if (lhs_size < rhs_size) { in SharedCompareImpl()
1439 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1079 size_t lhs_size = lhs.size();
1081 if (lhs_size != rhs_size) return false;
/external/libtextclassifier/abseil-cpp/absl/strings/
Dcord.cc1410 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1412 if (lhs_size == rhs_size) { in SharedCompareImpl()
1413 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1415 if (lhs_size < rhs_size) { in SharedCompareImpl()
1416 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1238 size_t lhs_size = lhs.size();
1240 if (lhs_size != rhs_size) return false;
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/strings/
Dcord.cc1410 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1412 if (lhs_size == rhs_size) { in SharedCompareImpl()
1413 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1415 if (lhs_size < rhs_size) { in SharedCompareImpl()
1416 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1238 size_t lhs_size = lhs.size();
1240 if (lhs_size != rhs_size) return false;
/external/abseil-cpp/absl/strings/
Dcord.cc1433 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1435 if (lhs_size == rhs_size) { in SharedCompareImpl()
1436 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1438 if (lhs_size < rhs_size) { in SharedCompareImpl()
1439 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1079 size_t lhs_size = lhs.size();
1081 if (lhs_size != rhs_size) return false;
/external/angle/third_party/abseil-cpp/absl/strings/
Dcord.cc1374 size_t lhs_size = lhs.size(); in SharedCompareImpl() local
1376 if (lhs_size == rhs_size) { in SharedCompareImpl()
1377 return GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
1379 if (lhs_size < rhs_size) { in SharedCompareImpl()
1380 auto data_comp_res = GenericCompare<int>(lhs, rhs, lhs_size); in SharedCompareImpl()
Dcord.h1460 size_t lhs_size = lhs.size();
1462 if (lhs_size != rhs_size) return false;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc893 int64 lhs_size = in ProfitableToMakeDotOperandColumnMajor() local
896 if (lhs_size < kColumnMajorThresholdInBytes) { in ProfitableToMakeDotOperandColumnMajor()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc149 auto lhs_size = dot_lhs->shape().dimensions(lhs_concat_dim + 1); in GenNewWindow() local
151 wd->set_size(lhs_size); in GenNewWindow()
154 wd->set_size(lhs_size); in GenNewWindow()
157 wd->set_stride(std::max<int64>(1, lhs_size - 1)); in GenNewWindow()
159 wd->set_base_dilation(lhs_size); in GenNewWindow()