- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 36 for hasAttr (0.45 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
// MLIR. This will also require some trait to identify optional attributes in // MLIR. constexpr char kShapeInferenceGraph[] = "shape_inference_graph"; if (mlir::isa<mlir::TF::XlaHostComputeOp>(inst) && !inst->hasAttr(kShapeInferenceGraph) && !attrs_to_ignore.contains(kShapeInferenceGraph)) { AttrValue value; value.mutable_func()->set_name(""); (*attributes)[kShapeInferenceGraph] = value; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
quantizing_op, custom_map)) { if (!(enable_verify && enable_whole_model_verify)) { return failure(); } if (quantizing_op->hasAttr(kDebugModeOpQuantAttrName) || quantizing_op->hasAttr(kDebugModeOpFloatAttrName)) { return failure(); } rewriter.setInsertionPoint(quantizing_op); Operation* float_op = rewriter.clone(*quantizing_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
auto tf_dialect = ctx->getLoadedDialect<TF::TensorFlowDialect>(); func.walk([&](Operation *op) { if (op->getDialect() != tf_dialect) return; if (IsAllowListedOp(op)) return; if (op->hasAttr(kNoFallbackAttr)) { op->removeAttr(kNoFallbackAttr); return; } if (!ConvertToFlexOp(op)) signalPassFailure(); }); } } // namespace internal
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
const int operand_index) { auto* defining_op = op->getOperand(operand_index).getDefiningOp(); for (auto attr : kQuantizationAxisAttrs) { if (defining_op->hasAttr(attr)) { return defining_op->getAttr(attr); } } // Not found. return rewriter.getI64IntegerAttr(-1); } LogicalResult CheckIfAttrIs8Bit(const std::string& attr, Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
if (isa<TPUPartitionedCallOp>(call)) return false; if (isa<RemoteCallOp>(call)) return false; // Maintain inlining for `tf.function`s with jit_compile option. if (callable->hasAttr("tf._XlaMustCompile")) return true; auto noinline_attr_name = absl::StrCat("tf.", tensorflow::kNoInlineAttr); if (auto noinline_attr = callable->getAttrOfType<BoolAttr>(noinline_attr_name))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
op->getAttrOfType<ArrayAttr>(kTfSavedModelExportedNamesAttr); return exported_names && !exported_names.empty(); } // Check if a function is an entry function. bool IsEntryFunction(func::FuncOp op) { return op->hasAttr(kEntryFunctionAttr); } // Returns true iff the provided FuncOp is qualified to be included in the main // function. bool ShouldIncludeInMainFunction(func::FuncOp func_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
bool IsQuantizedXlaCallModuleOp(TF::XlaCallModuleOp xla_call_module_op) { return !IsOpNotQuantized(xla_call_module_op) && xla_call_module_op->hasAttr(kQuantTraitAttrName) && xla_call_module_op->hasAttr(kEntryFuncAttrName); } // Returns the entry function, i.e. the callee of `xla_call_module_op`. func::FuncOp GetEntryFuncOp(TF::XlaCallModuleOp xla_call_module_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
} if (op.hasAttr(mlir::TF::kReplicationInfoAttr)) { // For replicated case, borrow cluster structure from replication info. // Following condition is already checked in // `HasValidCompilationAndReplicationAttributes` above, assert here for // documentation and to avoid breakage when that function is changed. assert(op.hasAttr(mlir::TF::kCompileDeviceTypeAttr));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getSuccessors(), new_regions); inner_op = rewriter.create(state); } else { bool disable_call_shape_inference = false; if (op->hasAttr("_disable_call_shape_inference")) { disable_call_shape_inference = op->getAttrOfType<BoolAttr>("_disable_call_shape_inference") .getValue(); } inner_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/c/kernels.cc
const char* attr_name, TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); return cc_ctx->HasAttr(attr_name); } TF_StringView TF_OpKernelConstruction_GetName(TF_OpKernelConstruction* ctx) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelConstruction*>(ctx); TF_StringView string_view_of_name;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)