- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for TF_Op (0.04 sec)
-
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.h
bool operator()(TFFakeQuantOp tf_op, AttrType &min_value, AttrType &max_value) const { min_value = tf_op.getMinAttr(); max_value = tf_op.getMaxAttr(); return true; // Successfully matched and fetched. } }; template <class TFFakeQuantOp> struct FetchConstantMinMaxInputs { using AttrType = DenseFPElementsAttr; bool operator()(TFFakeQuantOp tf_op, AttrType &min_value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
} def TF_FlushSummaryWriterOp : TF_Op<"FlushSummaryWriter", []> { let summary = "Flushes the writer's unwritten events."; let description = [{ writer: A handle to the summary writer resource. }]; let arguments = (ins Arg<TF_ResourceTensor, "", [TF_SummaryWrite]>:$writer ); let results = (outs); } def TF_ImportEventOp : TF_Op<"ImportEvent", []> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/fake_quant_utils.h
bool operator()(TFFakeQuantOp tf_op, AttrType &min_value, AttrType &max_value) const { min_value = tf_op.getMinAttr(); max_value = tf_op.getMaxAttr(); return true; // Successfully matched and fetched. } }; template <class TFFakeQuantOp> struct FetchConstantMinMaxInputs { using AttrType = DenseFPElementsAttr; bool operator()(TFFakeQuantOp tf_op, AttrType &min_value,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
); } def TF_AnonymousMemoryCacheOp : TF_Op<"AnonymousMemoryCache", [TF_UniqueResourceAllocation]> { let summary = ""; let arguments = (ins); let results = (outs Res<TF_ResourceTensor, "", [TF_DatasetMemoryCacheAlloc]>:$handle, TF_VariantTensor:$deleter ); } def TF_AnonymousMultiDeviceIteratorOp : TF_Op<"AnonymousMultiDeviceIterator", [TF_UniqueResourceAllocation]> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
// attribute to create the quantization parameter for the new quantize op. rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType(); TypeAttr qtype = quant::GetQuantizedTypeAttr( rewriter, res_type, min_value, max_value, quant_dim, num_bits,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
} auto attr_value = mlir::DenseI32ArrayAttr::get(tf_op->getContext(), values); tf_op->setAttr(attr_name, attr_value); }; if (tf_op->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>() || tf_op->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { // The op has multiple variadic operands or results.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.td
include "mlir/Interfaces/CallInterfaces.td" include "mlir/Interfaces/InferTypeOpInterface.td" include "mlir/IR/OpAsmInterface.td" // TODO(b/230804708): Add hybrid dot general. def TF_UniformQuantizedDotGeneralOp : TF_Op<"UniformQuantizedDotGeneral", [Pure]> { // TODO(b/230804708): Improve the operator description. let summary = "Quantized dot general operator."; let arguments = (ins TensorOf<[TF_Float32, TF_Qint8]>:$lhs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.td
"$_builder.getStringAttr("# values #")">; def LegalizeTensorListReserve : Pat<(TF_TensorListReserveOp:$tf_op $shape, $num_elements), (TFL_CustomOp (Size2InputRange $shape, $num_elements), (CreateStringAttr<"\"TensorListReserve\"">), (CustomOptions $tf_op))>; def LegalizeTensorListStack : Pat<(TF_TensorListStackOp $input, $shape, $unused_num_elements), (TFL_CustomOp (Size2InputRange $input, $shape),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 18 07:12:51 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td
// implementation available only in TFRT. #ifndef TFRT_OPS #define TFRT_OPS include "tensorflow/compiler/mlir/tensorflow/ir/tf_op_base.td" include "mlir/IR/OpBase.td" def TF__TfrtSetResourceOp : TF_Op<"_TfrtSetResource", []> { let summary = "Set values as TFRT runtime's static resource."; let description = [{ Setting a tensor value in TFRT runtime's static resource manager, using
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
#include "tensorflow/compiler/mlir/lite/transforms/generated_legalize_tf.inc" #define DECL_CONVERT_OP(tf_op) \ struct ConvertTF##tf_op##Op : public RewritePattern { \ explicit ConvertTF##tf_op##Op(MLIRContext* context) \ : RewritePattern(TF::tf_op##Op::getOperationName(), 1, context) {} \
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0)