- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 69 for ConstantOp (0.16 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.h
// Duplicates the constant op if it has multiple uses, and replaces // target_op->operand[operand_index] with the newly created op. This also // replaces corresponsing quantization states. arith::ConstantOp DuplicateConstantOpIfNeeded(arith::ConstantOp op, Operation* target_op, int operand_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:42:17 UTC 2024 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_ops.h
REGISTER_KERNEL_BUILDER( \ Name("Const").Device(DEVICE).TypeConstraint("dtype", TYPES), \ ConstantOp); \ REGISTER_KERNEL_BUILDER( \ Name("Identity").Device(DEVICE).TypeConstraint("T", TYPES), IdentityOp); \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 23 19:28:25 UTC 2021 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" namespace tflite { namespace { bool IsConst(mlir::Operation* op) { return llvm::isa<mlir::arith::ConstantOp, mlir::TF::ConstOp, mlir::TFL::ConstOp, mlir::TFL::QConstOp>(op); } bool IsOpSupported(mlir::Operation* op, const std::string& hardware) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
tensorflow::GetTypeFromTFTensorShape(shape, rewriter->getIntegerType(32)); DenseElementsAttr attr = DenseElementsAttr::get(type, rewriter->getI32IntegerAttr(val)); return rewriter->create<arith::ConstantOp>(loc, type, attr); } Value CreateI64SplatConst(Location loc, PatternRewriter *rewriter, ArrayRef<int64_t> shape, int64_t val) { RankedTensorType type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
void DuplicateSmallConstantOps(ModuleOp module_op, func::FuncOp main_func) { OpBuilder builder(main_func.getContext()); for (auto constant_op : main_func.getBody().getOps<mlir::stablehlo::ConstantOp>()) { builder.setInsertionPointAfter(constant_op); if (constant_op.getResult().use_empty() || constant_op.getResult().hasOneUse()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
func->setAttr("tfl.func", builder.getUnitAttr()); builder.setInsertionPointToStart(func.addEntryBlock()); llvm::SmallVector<int> shape{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; builder.create<mlir::arith::ConstantOp>( builder.getUnknownLoc(), mlir::DenseIntElementsAttr::get( mlir::RankedTensorType::get(shape.size(), builder.getI32Type()), shape));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
{pad_op.getEdgePaddingLow().size()}, rewriter.getI64Type()); auto slice_begins_const_op = rewriter.create<arith::ConstantOp>( loc, slice_attr_type, DenseIntElementsAttr::get(slice_attr_type, slice_begins)); auto slice_sizes_const_op = rewriter.create<arith::ConstantOp>( loc, slice_attr_type, DenseIntElementsAttr::get(slice_attr_type, slice_sizes));
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/transforms/optimize_batch_matmul.cc
// mapped to X and Z dimension. std::iter_swap(permute.begin() + input_rank - 1, permute.begin() + input_rank - 2); auto permutation_tensor_op = rewriter.create<arith::ConstantOp>( bmm_op->getLoc(), permuation_tensor_type, DenseElementsAttr::get(permuation_tensor_type, permute)); auto input_shape = input_type.getShape();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
// output gate bias is 0 since it is out of bounds of the bias tensor, so // we set its value as a const tensor of specified size and value 0. EXPECT_TRUE( mlir::cast<ElementsAttr>(mlir::cast<mlir::arith::ConstantOp>( it->getOpOperand(15).get().getDefiningOp()) .getValue()) .getValues<FloatAttr>()[0] .getValue()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
Value input = op.getOperand(index); if (input.getDefiningOp() == nullptr) continue; // TODO(b/172517537): make this work with non-PTQ case. if (llvm::isa<func::ConstantOp, arith::ConstantOp, TFL::ConstOp>( input.getDefiningOp())) { // Tensors with derived scale are biases, and handled in propagation. if (tensor_property.use_derived_scale) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0)