- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 128 for get_shape (0.26 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
// Get the new shape. SmallVector<int64_t, 4> new_shape; for (int i = 0; i < 4 - rank; ++i) { new_shape.push_back(1); } for (auto size : input_type.getShape()) { new_shape.push_back(size); } auto reshape_op = InsertReshapeOp(concat_op.getLoc(), input, input_type.getElementType(), new_shape, &rewriter); reshape_ops.push_back(reshape_op.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
mlir::cast<ShapedType>(lhs.getType()).getShape(), mlir::cast<ShapedType>(rhs.getType()).getShape(), symbolic_broadcast_shape)) { return failure(); } // Calculates the broadcast shape using BroadcastArgs op. Value lhs_shape = GetShape(lhs, op->getLoc(), rewriter); Value rhs_shape = GetShape(rhs, op->getLoc(), rewriter); auto broadcast_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
TF::ReshapeOp GetReshapeOpForConv2DFilter(ArrayRef<int64_t> new_shape, Value input, OpBuilder* builder) { auto reshape_result_type = RankedTensorType::get(new_shape, getElementTypeOrSelf(input)); auto reshape_type = RankedTensorType::get( {static_cast<int64_t>(new_shape.size())}, builder->getIntegerType(64));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_tensor_helper.cc
} } ArrayRef<int64_t> shape = ranked_ty.getShape(); SmallVector<int64_t, 4> out_shape; out_shape.reserve(rank - (keep_dims.getValue() ? 0 : num_reduce_dim)); for (int64_t i = 0; i < rank; ++i) { if (!is_reduce_dim[i]) out_shape.push_back(shape[i]); else if (keep_dims.getValue()) out_shape.push_back(1); } return RankedTensorType::get(out_shape, element_ty); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
ArrayRef<int64_t> shape = ranked_type.getShape(); assert(permutation.size() == shape.size()); SmallVector<int64_t, 4> new_shape(permutation.size()); for (size_t i = 0; i < permutation.size(); ++i) new_shape[i] = shape[permutation[i]]; return RankedTensorType::get(new_shape, ranked_type.getElementType()); } return type; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
const int64_t b = lhs_type.getShape()[std::get<0>(i)]; lhs_shape.push_back(b); out_shape->push_back(b); lhs_reshape_segids.at(std::get<0>(i)) = num_lhs_reshape_segids++; } } else { const int64_t lhs_out_size = ProdShapeWithIndexInTuple<0>(lhs_type.getShape(), dnums.lhs_out); lhs_shape.push_back(lhs_out_size); out_shape->push_back(lhs_out_size);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
llvm::ArrayRef<int32_t> perm) { llvm::SmallVector<int64_t> new_shape(shape.size()); for (const auto &perm_element : enumerate(perm)) { new_shape[perm_element.index()] = shape[perm_element.value()]; } return new_shape; } // Determine if op commutes with transposes. Requires a strict // definition of Elementwise, all i/o shapes and types must be same-rank
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
auto new_shape_const_attr = DenseElementsAttr::get(shape_spec_type, new_shape.getShape()); rewriter.setInsertionPointAfter(weight_op); auto new_shape_const = rewriter.create<arith::ConstantOp>( weight_op->getLoc(), shape_spec_type, new_shape_const_attr); auto reshape_op = rewriter.create<TF::ReshapeOp>( weight_op->getLoc(), new_shape, weight_op->getResult(0), new_shape_const);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
auto element_type = x.getElementType(); auto shape_x = x.getShape(); auto shape_y = y.getShape(); if (shape_x.size() == shape_y.size()) { llvm::SmallVector<int64_t, 4> out_shape(shape_x.size()); for (int i = 0; i < shape_x.size(); i++) { auto x_val = shape_x[i]; auto y_val = shape_y[i]; out_shape[i] = std::max(x_val, y_val); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
value_attr.getValues<Attribute>()[outer_index * inner_dims_size + i]); } } std::vector<int64_t> new_shape = value_attr.getType().getShape().vec(); new_shape[0] = lookup_attr.getType().getShape()[0]; auto new_type = value_attr.getType().clone(new_shape); return DenseElementsAttr::get(new_type, new_values); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)