- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for SmallVector (0.21 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// TF::UnsortedSegmentProdOp. llvm::SmallVector<int32_t, 4> flattened_out_segids = llvm::SmallVector<int32_t, 4>(operand_rank, static_cast<int32_t>(-1)); for (int64_t i : dot_dimensions_info.out_dimensions().AxesArray()) { flattened_out_segids[i] = 0; } llvm::SmallVector<int32_t, 4> flattened_contracting_segids = llvm::SmallVector<int32_t, 4>(operand_rank, static_cast<int32_t>(-1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
ArrayRef<int64_t> filter_shape = filter_value_attr.getShapedType().getShape(); SmallVector<int8_t> filter_constant_values{ filter_value_attr.getValues<int8_t>()}; SmallVector<int8_t> new_filter_constant_values(filter_constant_values.size(), 0); SmallVector<int64_t, 4> transpose_dims; if (is_depthwise) { transpose_dims = {2, 0, 1, 3}; } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} std::optional<RankedTensorType> InferXlaConvOutputShape( llvm::SmallVector<int64_t> input_tensor_dims, llvm::SmallVector<int64_t> kernel_tensor_dims, llvm::SmallVector<int64_t> window_strides, llvm::SmallVector<std::pair<int64_t, int64_t>> paddings, llvm::SmallVector<int64_t> lhs_dilations, llvm::SmallVector<int64_t> rhs_dilations, int64_t batch_group_count,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// shapes. LogicalResult BroadcastGradientArgsOp::verify() { BroadcastGradientArgsOp op = *this; SmallVector<int64_t, 4> s0_shape, s1_shape; DenseIntElementsAttr s0, s1; if (!ExtractInputConstShape(op, s0, s1, s0_shape, s1_shape)) return success(); // If both shape is known const, try to validate shape on them as well. SmallVector<int64_t, 4> bcasted_shape; if (!OpTrait::util::getBroadcastedShape(s0_shape, s1_shape, bcasted_shape))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
} } // The While ops operands and result types need to match SmallVector<Value, 4> new_operands; SmallVector<Value, 4> new_body_yield; SmallVector<bool, 4> removed_operand(while_op.getNumOperands(), false); llvm::SmallVector<Type, 4> types; new_operands.reserve(while_op.getNumOperands()); new_body_yield.reserve(while_op.getNumOperands());
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/tensorflow/ir/tf_ops_n_z.cc
return success(); } int width = getElementTypeOrSelf(op.getType(0)).getIntOrFloatBitWidth(); SmallVector<Value, 4> results(op.getNumOperands()); SmallVector<int64_t, 4> dynamic_indices; SmallVector<Value, 4> dynamic_inputs; SmallVector<Type, 4> result_types; for (const auto &e : llvm::enumerate(op.getOperands())) {
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/transforms/optimize.cc
DenseIntElementsAttr perm_const; if (!matchPattern(perm, m_Constant(&perm_const))) return false; SmallVector<int64_t, 4> axes; for (const auto &axis_int : perm_const.getValues<APInt>()) { axes.push_back(axis_int.getSExtValue()); } return (axes == SmallVector<int64_t>({0, 3, 1, 2})); } #include "tensorflow/compiler/mlir/lite/transforms/generated_optimize.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
const absl::InlinedVector<Node*, 4>& control_ret_nodes) { // Store the arg/return attributes as a list rather than uniqueuing during // construction. llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs; arg_attrs.resize(func.getNumArguments()); llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs; ret_attrs.resize(func.getNumResults()); auto set_attributes_on_func = [&](Node* node, int64_t index, bool is_arg) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto dict_attr = fn->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); if (!dict_attr) return; llvm::SmallVector<llvm::StringRef, 2> input_names; llvm::SmallVector<llvm::StringRef, 2> output_names; if (auto str = mlir::dyn_cast_or_null<mlir::StringAttr>(dict_attr.get("inputs"))) { str.getValue().split(input_names, ',', /*MaxSplit=*/-1,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)