- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getI32TensorAttr (0.12 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
RankedTensorType::get(static_cast<int32_t>(1), builder.getI32Type()); auto out_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_out_segids)); auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor = builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
llvm::SmallVector<int32_t> const_i32_vec; for (auto element : const_value) { const_i32_vec.push_back(static_cast<int32_t>(element)); } const_value_raw = rewriter.getI32TensorAttr(const_i32_vec); } Value result_const = builder.create<TF::ConstOp>(const_value_raw); return result_const; } llvm::SmallVector<int64_t> GetInversePermutationArray(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
auto values_type = RankedTensorType::get( {static_cast<int32_t>(values.size())}, rewriter->getIntegerType(32)); auto constant_attr = rewriter->getI32TensorAttr(values); return rewriter->create<arith::ConstantOp>(loc, values_type, constant_attr); } // Creates ConstantOp for array of int64_t. arith::ConstantOp createI64ConstantOp(llvm::ArrayRef<int64_t> values,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0)