- Sort Score
- Result 10 results
- Languages All
Results 151 - 156 of 156 for RankedTensorType (0.64 sec)
-
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
quant::UniformQuantizedPerAxisType ResetAxisAndBroadcast( const ArrayRef<int64_t> shape, const quant::UniformQuantizedPerAxisType qtype, const Type target, const int quant_dim) { const auto shaped = dyn_cast<RankedTensorType>(target); if (!shaped) return {}; const ArrayRef<int64_t> new_shape = shaped.getShape(); SmallVector<double, 4> scales(qtype.getScales().begin(), qtype.getScales().end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
/*scale=*/1.0 / output_inverse_scale_value, output_zero_point_value); Value conv_output_value = op.getResult(); auto output_uniform_quantized_tensor_type = RankedTensorType::getChecked( rewriter.getUnknownLoc(), /*shape=*/ mlir::cast<TensorType>(conv_output_value.getType()).getShape(), output_uniform_quantized_type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
builder.setInsertionPoint(op); if (!const_true) const_true = builder.create<TF::ConstOp>( op.getLoc(), DenseIntElementsAttr::get( RankedTensorType::get(/*shape=*/{}, builder.getI1Type()), true)); op.getIsInitialized().replaceAllUsesWith(const_true); op.erase(); } } // Performs store-load forwarding. This effectively removes
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
CPred<"$_op.getOperand(" # n # ").getType().isa<RankedTensorType>()">, CPred<"$_op.getOperand(" # n # ").getType().cast<ShapedType>().getRank() == " # dim>]>; // True if operand n is ranked and has a rank > dim. class TFL_OperandIsRankedAndHasDimPred<int n, int dim> : And<[ CPred<"$_op.getOperand(" # n # ").getType().isa<RankedTensorType>()">,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
constexpr bool kDoAssertions = true; void AddAssertion(OpBuilder& builder, Location& loc, Value cond, const std::string& message) { if (!kDoAssertions) return; auto shape_type = RankedTensorType::get({1}, builder.getType<TF::StringType>()); auto msg = builder.create<TF::ConstOp>( loc, DenseStringElementsAttr::get(shape_type, llvm::ArrayRef<StringRef>{message}));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
OpBuilder<(ins "StringAttr":$container, "StringAttr":$shared_name, "BoolAttr":$use_node_name_sharing, "TypeAttr":$key_dtype, "TypeAttr":$value_dtype), [{ build($_builder, $_state, mlir::RankedTensorType::get({}, $_builder.getType<mlir::TF::ResourceType>()), container, shared_name, use_node_name_sharing, key_dtype, value_dtype); }]>]; } def TF_IFFTOp : TF_Op<"IFFT", [Pure]> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)