- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 61 for if_builder (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/optimize.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) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
shape_determination_fns, compilation_result); } static mlir::LogicalResult MlirTfToHloTextTranslateFunctionImpl( mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) { if (!module_op) return mlir::failure(); llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; auto args_status = ParseArgumentShapes(mlir::StringRefToView(input_shapes), arg_shapes);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
`tf_device` dialect operation. }]; let arguments = (ins Variadic<AnyType>:$results ); let builders = [ OpBuilder<(ins), [{ build($_builder, $_state, {}); }]> ]; let assemblyFormat = "attr-dict ($results^ `:` type($results))?"; } def TfDevice_LaunchFuncOp : TfDevice_Op<"launch_func", []> { let summary = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
tensors. Each tensor contains the asset file names. """ v1_builder = builder.SavedModelBuilder(saved_model_path) sig_def = signature_def_utils_impl.predict_signature_def( inputs=inputs, outputs=outputs ) v1_builder.add_meta_graph_and_variables( sess, tags, signature_def_map={signature_key: sig_def},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 08:51:46 UTC 2024 - 51.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
(HasNonNegativeOffset $begin, $end)]>; def GetNumElementsOrOne: NativeCodeCall<"GetNumElementsOrOne($0.getType())">; def ReshapeValueDroppingLastDim : NativeCodeCall< "ReshapeValueDroppingLastDim($_builder, $0)">; def IsOneHotIndexAttribute : Constraint<CPred< "TFL::IsOneHotIndexAttribute($0)">>; // Checks if the shape has static shape with last dimension equals 1.
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/lite/ir/tfl_ops.td
"StringAttr":$fusedActivationFunction), [{ buildFusedBroadcastableBinOp( &$_builder, $_state, lhs, rhs, fusedActivationFunction); }]>; def TFL_ComparisonBinaryBuilder : OpBuilder<(ins "Value":$lhs, "Value":$rhs), [{ buildComparisonBinOp(&$_builder, $_state, lhs, rhs); }]>; //===----------------------------------------------------------------------===//
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/tfr/ir/tfr_ops.td
// tensor argument types class TFR_Type<string name> : DialectType<TFR_Dialect, CPred<"$_self.isa<mlir::TFR::" # name # "Type>()">, "TFR " # name #" type">, BuildableType<"$_builder.getType<mlir::TFR::" # name # "Type>()">; def TFR_TensorType : TFR_Type<"TFRTensor">; def TFR_TensorListType : TFR_Type<"TFRTensorList">; def TFR_AllTensorTypes : Type<Or<[ TFR_TensorType.predicate,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
) signature_map = {'main': signature} tags = {tag_constants.SERVING} v1_builder = builder.SavedModelBuilder(self._input_saved_model_path) v1_builder.add_meta_graph_and_variables( sess, tags, signature_def_map=signature_map ) v1_builder.save() quantization_options = quant_opts_pb2.QuantizationOptions(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
src/cmd/go/internal/work/exec.go
} if argLen > sys.ExecArgLengthLimit { return true } // On the Go build system, use response files about 10% of the // time, just to exercise this codepath. isBuilder := os.Getenv("GO_BUILDER_NAME") != "" if isBuilder && rand.Intn(10) == 0 { return true } return false } // encodeArg encodes an argument for response file writing. func encodeArg(arg string) string {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Mon Jun 10 14:46:37 UTC 2024 - 105.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
"llvm::ArrayRef<::mlir::NamedAttribute>":$attributes, "unsigned":$numRegions), [{ assert(numRegions == 2u && "mismatched number of regions"); build($_builder, $_state, resultTypes, operands, attributes); }]>]; let hasCanonicalizer = 1; } def TF_GeneratorDatasetRegionOp : TF_Op<"GeneratorDatasetRegion", [AttrSizedOperandSegments,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)