- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 73 for BoolAttr (0.54 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
Variadic<TF_Tensor>:$input, ConfinedAttr<SymbolRefArrayAttr, [ArrayMinCount<1>]>:$branches, // Used to map StatelessCase and Case op defined in TensorFlow to a common // op. BoolAttr:$is_stateless ); let results = (outs Variadic<TF_Tensor>:$output ); TF_DerivedOperandTypeListAttr Tin = TF_DerivedOperandTypeListAttr<1>;
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/tensorflow/ir/tf_ops_tensor_helper.cc
// incompatible shapes. Otherwise, returns a tensor type with unknown rank. Type DeduceEqualCmpOpType(Builder *builder, Location loc, Value x, Value y, BoolAttr incompatible_shape_error) { auto result_type = OpTrait::util::getBroadcastedType(x.getType(), y.getType()); if (!result_type) { if (incompatible_shape_error.getValue()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
"whose value is non-negative">; def TFL_BoolTrue : AttrConstraint< CPred<"$_self.isa<BoolAttr>() && $_self.cast<BoolAttr>().getValue()">, "whose value is true">; def TFL_BoolFalse : AttrConstraint< CPred<"$_self.isa<BoolAttr>() && !$_self.cast<BoolAttr>().getValue()">, "whose value is false">; class TFL_StringEqualsTo<string value> : AttrConstraint<
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/host_runtime/tpu_metadata_utils.cc
// Note: this information is duplicated and can be removed from the proto // and here once MLIR bridge phase 2 doesn't fallback to the old bridge. auto attr = op.getFuncOp().getArgAttrOfType<mlir::BoolAttr>( index, replication_attr_name); arg->set_is_same_data_across_replicas(attr != nullptr && attr.getValue()); // Currently only support first dimension to be bounded dynamic.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
TF_RETURN_IF_ERROR(ConvertLocation(locations[i], node_name, debug_info)); } } return absl::OkStatus(); } Status ConvertAttribute(const mlir::BoolAttr& attr, AttrValue* value) { value->set_b(attr.getValue()); return absl::OkStatus(); } Status ConvertAttribute(const mlir::IntegerAttr& attr, AttrValue* value) { value->set_i(attr.getInt());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
// If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) { if (!mlir::cast<BoolAttr>(legalize_tfl_variables_attr).getValue()) return; } RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); (void)applyPatternsAndFoldGreedily(module, std::move(patterns)); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
std::optional<bool> GetBoolFromCompositeAttr( const DictionaryAttr& composite_attrs, llvm::StringRef attr_name) { auto attr = composite_attrs.get(attr_name); if (!attr) return std::nullopt; if (auto bool_attr = mlir::dyn_cast_or_null<BoolAttr>(attr)) { return bool_attr.getValue(); } return std::nullopt; } ShapedType GetNhwcReturnTypeFromNchw(Operation* old_op) { auto composite_result_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
} // Convert `_XlaMustCompile`. if (op->hasAttr(tensorflow::kMustCompileAttr)) { bool must_compile_attr_val = op->getAttrOfType<BoolAttr>(tensorflow::kMustCompileAttr).getValue(); op->removeAttr(tensorflow::kMustCompileAttr); if (!must_compile_attr_val) { if (op->hasAttr(tensorflow::kCompileDeviceTypeAttr)) { op->emitOpError()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.td
} def TF_DumpTensorOp : TF_Op<"DumpTensor", []> { let summary = "Dump tensor proto."; let arguments = (ins Arg<TF_Tensor>:$input, StrAttr:$log_dir_path, StrAttr:$file_name, BoolAttr:$enabled, StrAttr:$func_name, StrAttr:$node_name ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
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/tensorflow/transforms/mark_initialized_variables.cc
is_variable_initialized = IsVariableInitialized(var_op, handle.device(), mgr, session); } var_op->setAttr("_is_initialized", BoolAttr::get(context, is_variable_initialized)); } return success(); } LogicalResult MarkInitializedVariablesInFunction(ModuleOp module, tensorflow::Session* session) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 19:14:56 UTC 2023 - 4.1K bytes - Viewed (0)