- Sort Score
- Result 10 results
- Languages All
Results 21 - 27 of 27 for NativeCodeCall (0.53 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.td
include "mlir/IR/PatternBase.td" include "tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td" // Combines the two variadic arguments ($in_tensors and $captured_tensors). def GetBatchFunctionOpArgOperands: NativeCodeCall<"cast<TF::BatchFunctionOp>($0[0].getDefiningOp()).getArgOperands()">; // Replaces `TF_BatchFunctionOp` into `TF_PartitionedCallOp` that calls the // same $f. This may be required, for example, when inlining is desired,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 18:58:35 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_data_optimization.td
include "mlir/IR/OpBase.td" include "mlir/IR/PatternBase.td" include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" // TODO(jpienaar): Move this somewhere general. class GetI64ScalarElementsAttr<int value> : NativeCodeCall<"DenseElementsAttr::get<int64_t>(RankedTensorType::get({}, $_builder.getIntegerType(64)), " # value # ")">; def FuseMapAndBatch : Pat< (TF_BatchDatasetV2Op
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
$axis), [(F32ElementsVal $output)]>; def Get1DShapeValue: NativeCodeCall<"Get1DShapeValue($_builder, $0)">; class GetIthValue<int index> : NativeCodeCall<"$0[" # index # "]">; def GetEmbeddingLookupShape: NativeCodeCall<"GetEmbeddingLookupShape($0, $1)">; // Replace // OneHot(index, depth, on=1.0f, off=0.0f, axis=-1) * filter // With
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose_patterns.td
include "mlir/IR/PatternBase.td" include "mlir/Dialect/Arith/IR/ArithOps.td" include "mlir/Dialect/Func/IR/FuncOps.td" include "tensorflow/compiler/mlir/tfr/ir/tfr_ops.td" class Quantize<string value> : NativeCodeCall<"TFR::Quantize(" # value # ", $0, $1, $_builder)">; class HasStringAttr<string value> : AttrConstraint< CPred<"$_self.cast<StringAttr>().getValue() == \"" # value # "\"">>; def QuantActRangeNonePattern :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_patterns.td
include "tensorflow/compiler/mlir/lite/ir/tfl_ops.td" // Quantize attribute $0 by using quantization parameter from %1. def QuantizeByQuantizedType : NativeCodeCall<"quant::Quantize($0, $1.getValue())">; def F32ElementsAttr : ElementsAttrBase< CPred<"$_self.cast<ElementsAttr>().getShapedType().getElementType().isF32()">, "float constant tensor">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:10:13 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_patterns.td
include "mlir/Dialect/Func/IR/FuncOps.td" include "mlir/Dialect/Arith/IR/ArithOps.td" include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" // Marks the op as no fallback. def MarkNoFallback : NativeCodeCall<"SetNoFallbackAttr($_builder, $0)">; def NoFallbackAttrNotSet : Constraint<CPred< "!$0.getDefiningOp()->hasAttr(kNoFallbackAttr)">>; class FloatValueEquals<string val> : Constraint<CPred<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.td
include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td" def IsDataFormatNHWC : ConstantAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">; // Get the last dimension size as a 1-d single element attr. def GetLastDimSizeAsI32 : NativeCodeCall< "DenseElementsAttr::get(RankedTensorType::get({1}, $_builder.getIntegerType(32)), " "static_cast<int32_t>($0.getType().cast<RankedTensorType>().getDimSize( " " $0.getType().cast<RankedTensorType>().getRank() - 1)))">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 22 07:31:23 UTC 2023 - 5.4K bytes - Viewed (0)