- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 25 for constant_ops (0.24 sec)
-
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/representative_dataset_test.py
from tensorflow.core.protobuf import meta_graph_pb2 from tensorflow.python.client import session from tensorflow.python.framework import constant_op from tensorflow.python.framework import ops from tensorflow.python.framework import test_util from tensorflow.python.platform import test from tensorflow.python.types import core
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 04 07:35:19 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
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, Location loc, PatternRewriter* rewriter) { auto values_type = RankedTensorType::get(
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/utils/lstm_utils.cc
return builder->create<arith::ConstantOp>(location, type, attr); } Value CreateF32SplatConst(OpBuilder* builder, ArrayRef<int64_t> shape, float val, mlir::Location location) { auto type = RankedTensorType::get(shape, builder->getF32Type()); auto attr = DenseElementsAttr::get(type, val); return builder->create<arith::ConstantOp>(location, type, attr); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// It is legal to have TF ops in the graph still which can be // used later or in the case of SELECT where we allow TF ops in the final // graph. target.addLegalOp<mlir::arith::ConstantOp>(); target.addLegalOp<mlir::func::ConstantOp>(); target.addLegalOp<TFL::NoValueOp>(); target.addLegalOp<ConstOp>(); target.addLegalOp<DequantizeOp>(); target.addLegalOp<QConstOp>(); if (run_tfl_runtime_verification_) {
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/tfr/passes/decompose.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
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/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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfuse_batch_norm_pass.cc
} auto scalar_type = RankedTensorType::get(/*shape=*/{}, fp_type); auto epsilon_tensor_attr = DenseElementsAttr::get( scalar_type, {mlir::cast<Attribute>(epsilon_attr)}); Value epsilon = b.create<mhlo::ConstantOp>(epsilon_tensor_attr); auto dims_type = RankedTensorType::get(/*shape=*/{0}, b.getIntegerType(64)); auto dims = DenseIntElementsAttr::get(dims_type, SmallVector<int64_t, 1>{}); if (broadcast_to_type.hasStaticShape()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.2K bytes - Viewed (0)