- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for rhs_dims (0.1 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
SmallVector<int64_t> new_shape; for (auto idx : llvm::reverse(llvm::seq<int32_t>(0, rhs_shape.getRank()))) { const int64_t rhs_dim = rhs_shape.getDimSize(idx); if (num_elements % rhs_dim != 0) { return {}; } new_shape.push_back(rhs_dim); num_elements = num_elements / rhs_dim; if (num_elements == 1) break; } absl::c_reverse(new_shape); auto reshape_op = builder.create<TF::ReshapeOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
llvm::SmallVector<int64_t, 4> dims; dims.reserve(lhs.getRank()); for (auto dim : llvm::zip(lhs.getShape(), rhs.getShape())) { int64_t lhs_dim = std::get<0>(dim); if (lhs_dim == std::get<1>(dim)) { dims.push_back(lhs_dim); } else { dims.push_back(ShapedType::kDynamic); } } return tensorflow::GetTypeFromTFTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)