- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for output_ty_shape (0.56 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
bool shape_ty_zero_dim = false; int unknown_index = -1; // The product of constant shape argument excluding unknown dimension. int64_t shape_ty_size = 1; llvm::SmallVector<int64_t, 8> output_ty_shape; output_ty_shape.reserve(shape_attr.getNumElements()); for (const auto &dim : llvm::enumerate(shape_attr.getValues<APInt>())) { const int64_t size = dim.value().getSExtValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
bool shape_ty_zero_dim = false; int unknown_index = -1; // The product of constant shape argument excluding unknown dimension. int64_t shape_ty_size = 1; llvm::SmallVector<int64_t, 8> output_ty_shape; output_ty_shape.reserve(shape_attr.getNumElements()); for (const auto& dim : llvm::enumerate(shape_attr.getValues<APInt>())) { const int64_t size = dim.value().getSExtValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto output_type = mlir::cast<ShapedType>(conv_op->getResult(0).getType()); SmallVector<int64_t, 4> output_2d_shape; for (int64_t dim : output_type.getShape()) { output_2d_shape.push_back(dim); } output_2d_shape.push_back(1); auto output_2d_type = RankedTensorType::get(output_2d_shape, output_type.getElementType()); SmallVector<int64_t, 4> output_permutation = {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)